JDAI-CV / centerX

This repo is implemented based on detectron2 and centernet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

转caffe遇到问题

starsky68 opened this issue · comments

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'

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

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

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

好的,谢谢回复。