xxradon / PytorchToCaffe

Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some bugs
MIT License
782 stars 224 forks source link

目标检测centernet里面用到DCNv2结构,您的工程不支持这个层转为caffe,请问可以添加此新层吗?谢谢! #30

Open zoufangyu1987 opened 5 years ago

zoufangyu1987 commented 5 years ago

目标检测centernet里面用到DCNv2结构,您的工程不支持这个层转为caffe,请问可以添加此新层吗?谢谢! DCNv2 structure is used in object detection CenterNet. Your project does not support the conversion of this layer to caffe. Can you add this new layer? Thank you!

xxradon commented 5 years ago

目前DCNV2在caffe里面没有对应的实现,所以无法添加。

zoufangyu1987 commented 5 years ago

这里有dcnv2资源,你可以看看,谢谢! https://github.com/zhanglonghao1992/Deformable-ConvNets-v1-v2-caffe

xxradon commented 5 years ago

目前来看,pytorch的DCNV2的参数和caffe的DCN传参有很大的区别,所以我这边推荐你使用pytorch 的C++API而不是caffe去实现centerNet的部署。

zoufangyu1987 commented 5 years ago

知道您的意思,谢谢!