zyayoung / Iter-Deformable-DETR

[CVPR2022] "Progressive End-to-End Object Detection in Crowded Scenes" on Deformable-DETR.
28 stars 6 forks source link

while runing main.py raise error"File "/tmp/pycharm_project_DETR/detr/deformable_transformer.py", line 392, in forward class_logits = self.class_embed[lid](output) TypeError: 'NoneType' object is not subscriptable" #3

Closed YuemingMrdream closed 2 years ago

zyayoung commented 2 years ago

We only support Deformable DETR with box refinement. Please make sure to include --with_box_refine in the arguments.