上传者: 42111465
|
上传时间: 2022-01-12 11:11:04
|
文件大小: 36KB
|
文件类型: -
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