IDEA-Research / DN-DETR

[CVPR 2022 Oral] Official implementation of DN-DETR
Apache License 2.0
530 stars 57 forks source link

Segmentation head #14

Closed K-Mike closed 2 years ago

K-Mike commented 2 years ago

Hi,

Please, tell me how to run your code with segmentation head? --masks - doesn't work

FengLi-ust commented 2 years ago

DN-DETR aims at object detection. We have not tested the code of DETRsegm, hence there is no quality assurance of these codes.

K-Mike commented 2 years ago

Could you give me the code? I can fix for my-self, and maybe help you to test and refactor the code.

FengLi-ust commented 2 years ago

We have released all the code in DN-DETR and do not implement a segmentation head. You can try to adapt the original DETR segmentation head to DN-DETR for segmentation tasks.