Deformable Convolution V2 PyTorch:PyTorch中的可变形ConvNets V2(DCNv2) 源码
PyTorch中的Deformable-ConvNets-V2 此存储库是的实现。 从原始移植。 有关完整框架,请参考此。 在可以找到基于mmdetection代码库的DCNv2的结果。 非常感谢强大而干净的框架。 master分支中的运算符与pytorch_v0.4.1兼容。 对于pytorch v1.0.0(由Jiajiai 实现)上的运算符,请参阅。 多亏了和mmlab的其他贡献者,DCNv2现在已基于该分支的主分支纳入正式的mmdetection回购中。 现在,它是使用新的cpp扩展api编写的,并且支持PyTorch 0.4.1和1.0,但速度和内存优化较小。 结果和模型可以在找到。 建造 sh make.sh 有关用法示例,请参见test.py和test_modulated.py 。 注意 此仓库提供了可变形的conv层,该层可在Deformable ConvNets
文件列表
Deformable-Convolution-V2-PyTorch-master.zip
(预估有个28文件)
Deformable-Convolution-V2-PyTorch-master
.gitignore
29B
functions
__init__.py
151B
modulated_dcn_func.py
7KB
deform_conv.py
5KB
test_modulated.py
4KB
src
deform_conv.c
565B
modulated_dcn_cuda.c
17KB
暂无评论