GHM_Detection icon indicating copy to clipboard operation
GHM_Detection copied to clipboard

TypeError: forward_train() got an unexpected keyword argument 'gt_bboxes_ignore

Open qilicun opened this issue 6 years ago • 6 comments

Hi,

Firstly, thank you very much for this work, great job! Secondly, when I tried to run this detection follow the introduction with VOC dataset, there was an error TypeError: forward_train() got an unexpected keyword argument 'gt_bboxes_ignore I modified cfg_retinanet_ghm_r50_fpn_1x.py, replace the data config part with that in GHM_Detection/mmdetection/configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py, what did I miss?

Thirdly, do you have any plan to release your pretrained model? Thanks again!

qilicun avatar Jan 26 '19 12:01 qilicun

Thank you for your affirmation.

It seems you still keep some configs of COCO, because the ignore region is used in COCO with the crowd label but not in VOC.

If you still have no clue, you can send an email to me with your config file and tell me which line the error locates. But since I am attending the AAAI conference now, maybe I will reply you a few days later.

libuyu avatar Jan 28 '19 08:01 libuyu

@libuyu Thanks for your reply. After I comment the setting with_crowd=False in train and val dict, it finally works!

Do you have any suggestion or tutorial that if I use the prtrained model of GHM with COCO datasets to train my own data?

qilicun avatar Jan 28 '19 09:01 qilicun

Sorry, I forgot that. I just use the official model. https://download.pytorch.org/models/resnet50-19c8e357.pth

libuyu avatar Jan 28 '19 10:01 libuyu

The model is trained on image-net and in theory, it can be used for fine-tuning on any data set.

libuyu avatar Jan 28 '19 10:01 libuyu

To be specific, my own detection data contains less than 1000 images, I wonder if trained with the pre-trained model on COCO would be better than pre-trained with the IMAGE-NET one?

qilicun avatar Jan 28 '19 10:01 qilicun

I understand your concern. But I think both COCO and ImageNet are very large compared with your set. So maybe the pre-trained model is not the key. Although you can study the effect of different pre-trained models. But in my opinion, how to avoid overfitting in such a small set is more important.

libuyu avatar Jan 30 '19 02:01 libuyu