search

Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
970 stars 105 forks source link

About the part of instance segmentaion #15

Closed GisRookie closed 1 year ago

GisRookie commented 1 year ago

Firstly, thanks for you great work, and i want to know about when the code of instance segmentation will release？

TempleX98 commented 1 year ago

We provide the config for Co-DETR with auxiliary mask branch: projects/configs/co_deformable_detr/co_deformable_detr_mask_r50_1x_coco.py. You can evaluate the roi_head after training.

GisRookie commented 1 year ago

Thank you very much!

GisRookie commented 1 year ago

How can i evaluate the roi head? when i run ./tools/dist_test.sh work_dir/co_deformable_detr_mask_r50_1x_coco.py work_dir/epoch_1.pth 3 --eval bbox segm error: in evaluate_det_segm raise KeyError(f'{metric} is not in results') KeyError: 'segm is not in results'

and i print the result_file: {'bbox': '/tmp/tmp75forie2/results.bbox.json', 'proposal': '/tmp/tmp75forie2/results.bbox.json'}

TempleX98 commented 1 year ago

You can set eval_module="two-stage" following this code

GisRookie commented 1 year ago

it works, thank you very much.