Hello, this is a really good job of yours. However, when I was replicating the code, I encountered the following problems. I saw that you also met similar problems, I would like to ask you how to deal with them, thank you very much!!
Traceback (most recent call last):
File "b4_quad.py", line 64, in
detect_quads(img_path,config_file=config_file)
File "/home/aiarhipov/centernet/post-processing/utils.py", line 143, in detect_quads
results = model(return_loss=False, rescale=True, **data)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/mmcv/runner/fp16_utils.py", line 116, in new_func
return old_func(*args, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/detectors/base.py", line 174, in forward
return self.forward_test(img, img_metas, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/detectors/base.py", line 147, in forward_test
return self.simple_test(imgs[0], img_metas[0], **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/detectors/single_stage.py", line 89, in simple_test
feat = self.extract_feat(img)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/detectors/single_stage.py", line 37, in extract_feat
x = self.backbone(img)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/backbones/dla_mmdet3d.py", line 474, in forward
x = getattr(self, "level{}".format(i))(x)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/backbones/dla_mmdet3d.py", line 283, in forward
x1 = self.tree1(x, identity)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/backbones/dla_mmdet3d.py", line 101, in forward
out += identity
RuntimeError: The size of tensor a (924) must match the size of tensor b (923) at non-singleton dimension 3
Hello, this is a really good job of yours. However, when I was replicating the code, I encountered the following problems. I saw that you also met similar problems, I would like to ask you how to deal with them, thank you very much!!
Traceback (most recent call last):
File "b4_quad.py", line 64, in
detect_quads(img_path,config_file=config_file)
File "/home/aiarhipov/centernet/post-processing/utils.py", line 143, in detect_quads
results = model(return_loss=False, rescale=True, **data)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/mmcv/runner/fp16_utils.py", line 116, in new_func
return old_func(*args, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/detectors/base.py", line 174, in forward
return self.forward_test(img, img_metas, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/detectors/base.py", line 147, in forward_test
return self.simple_test(imgs[0], img_metas[0], **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/detectors/single_stage.py", line 89, in simple_test
feat = self.extract_feat(img)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/detectors/single_stage.py", line 37, in extract_feat
x = self.backbone(img)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/backbones/dla_mmdet3d.py", line 474, in forward
x = getattr(self, "level{}".format(i))(x)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/backbones/dla_mmdet3d.py", line 283, in forward
x1 = self.tree1(x, identity)
File "/root/anaconda3/envs/openmmlab1/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/ctff/cqx/ocr/Cycle-CenterNet/mmdet/models/backbones/dla_mmdet3d.py", line 101, in forward
out += identity
RuntimeError: The size of tensor a (924) must match the size of tensor b (923) at non-singleton dimension 3