search

lucasjinreal / DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)
BSD 3-Clause "New" or "Revised" License
614 stars 125 forks source link

Error occurs when running testcpu.py #16

Open seekFire opened 3 years ago

seekFire commented 3 years ago

After building, when I run testcpu.py directly, the error occurs as follows:

Traceback (most recent call last): File "/home/bst/Documents/Projects/Official/Object_Detection/BBAVectors-Oriented-Object-Detection/models/DCNv2/testcpu.py", line 255, in <module> example_dconv() File "/home/bst/Documents/Projects/Official/Object_Detection/BBAVectors-Oriented-Object-Detection/models/DCNv2/testcpu.py", line 175, in example_dconv output = dcn(input) File "/home/bst/anaconda2/envs/Python3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/home/bst/Documents/Projects/Official/Object_Detection/BBAVectors-Oriented-Object-Detection/models/DCNv2/dcn_v2.py", line 191, in forward self.deformable_groups, File "/home/bst/Documents/Projects/Official/Object_Detection/BBAVectors-Oriented-Object-Detection/models/DCNv2/dcn_v2.py", line 39, in forward ctx.deformable_groups, IndexError: Dimension out of range (expected to be in range of [-2, 1], but got 2)

lucasjinreal commented 3 years ago

CPU am not tested actually. Please send PR if you got a chance to fix it.

BraveDrXuTF commented 3 years ago

I have the same problem

shitouren1994 commented 2 years ago

i have the same problem

songkq commented 2 years ago

the same issue