文件名称:Deformable-Convolution-V2-PyTorch:PyTorch中的可变形ConvNets V2(DCNv2)
文件大小:36KB
文件格式:ZIP
更新时间:2024-05-02 18:55:16
Cuda
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
----.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)
--------deform_conv_cuda.h(1KB)
--------modulated_dcn.c(1KB)
--------deform_conv.h(251B)
--------cuda()
--------modulated_dcn_cuda.h(3KB)
--------deform_conv_cuda_kernel.cu(26KB)
--------deform_conv_cuda.c(22KB)
--------deform_conv_cuda_kernel.h(2KB)
--------modulated_dcn.h(1KB)
----LICENSE(1KB)
----build_modulated.py(1KB)
----README.md(4KB)
----test.py(839B)
----build.py(943B)
----make.sh(438B)
----modules()
--------__init__.py(156B)
--------modulated_dcn.py(7KB)
--------deform_conv.py(1KB)