search

thangvubk / Cascade-RPN

Code for NeurIPS 2019 paper: "Cascade RPN: Delving into High-Quality Region Proposal Network with Adaptive Convolution"
Apache License 2.0
180 stars 19 forks source link

Question regarding bbox_overlaps.py #18

Open kartikJ-9 opened 4 years ago

kartikJ-9 commented 4 years ago

I am trying to implement Cascade RPN with the most recent mmdetection. I have a custom dataset with 4 different id for detection. After running train.py script, after the 1st epoch I am getting the following error:

" File "/home/user/Downloads/mmdetection-master/mmdet/core/evaluation/bbox_overlaps.py", line 33, in bbox_overlaps area2 = (bboxes2[:, 2] - bboxes2[:, 0] + 1) * ( IndexError: too many indices for array "

Is it because I am using a CustomDataset? What changes I should do to make it work for my own dataset?

I have a epoch_1.pth file generated but I am not able to get mAP chart as expected for the detection task. I am new to this project and any help with this would be appreciated .Thanks.

image

image

thangvubk commented 4 years ago

I suggest running on COCO dataset first. If you can run COCO dataset normally, then you can change to your own dataset more easily.