search

Epiphqny / SOLOv2

SOLOv2: Dynamic, Faster and Stronger, achives 39.5mAP on coco test-dev (36 epochs result)
https://arxiv.org/abs/2003.10152
Apache License 2.0
245 stars 37 forks source link

questions about testing my own data ? #18

Closed reedwiththought closed 4 years ago

reedwiththought commented 4 years ago

Hi @Epiphqny , I have successfully trained the model on my own dataset which has 5 categories('1', '2', '3', '4', '5'), and the test_ins_vis.py is ok, but the test_ins.py reports an error as follows:

image

After I print dataset.cat_ids[label], it can't be greater than 1, so I directly change data['category_id'] as follows and it's ok:

image

but it reports an another error as follows:

image

Could you give me some advice? thanks.

reedwiththought commented 4 years ago

I have soloved this problem. After replace it as follows and it' ok:

image