search

JDAI-CV / centerX

This repo is implemented based on detectron2 and centernet
Apache License 2.0
555 stars 86 forks source link

转caffe遇到问题 #43

Closed starsky68 closed 3 years ago

starsky68 commented 3 years ago

File "./modeling/layers/centernet_deconv.py", line 91, in forward offset = torch.cat((o1, o2), dim=1) File "/home/22/code/centerX-master/projects/speedup/pytorch_to_caffe.py", line 660, in call out = self.obj(self.raw, *args, **kwargs) TypeError: _cat() got an unexpected keyword argument 'dim'

配置文件和模型是匹配的，为何还会出现这个问题

CPFLAME commented 3 years ago

看上去你用了deformable conv，这个层要转码特别的麻烦，不建议转码时使用改层，建议直接换成普通的卷积。 refer to https://github.com/JDAI-CV/centerX/issues/36#issuecomment-842125288

starsky68 commented 3 years ago

看上去你用了deformable conv，这个层要转码特别的麻烦，不建议转码时使用改层，建议直接换成普通的卷积。 refer to #36 (comment)

好的，谢谢回复。