aliborji/DetectionUpperbound

Where can I get the json files?

Opened this issue · 1 comments

There are several json files such as Mask_LOC_IMG.json, coco_GT.json mentioned in Error_diagnosis.ipynb, so I'd like to know how to generate such json files? Thanks.

The only file that you need is the .json file which is the output of your detection model and the ground truth. Then you convert it to dictionary based on the images as keys using this:

GT_table = {}
for img in data_valid['images']:
img_bboxes = []
for bboxes in data_valid['annotations']:
if (bboxes['image_id']==img['id']):
img_bboxes.append(bboxes)
GT_table[img['id']]=img_bboxes

So basically "coco_GT.json" is the ground truth of coco after converting to the image based dictionary using the above code. You need this for speeding in the loop calls. Then you create each file step by step based on the code and explanation in the paper. For example, the "Mask_Ioc_IMG.json" is the output of the "mask-rcnn" detection after applying the localization step.