Yeah could anyone help this guy out ? I tried finding resources on how this could be done but failed to find code that explained how to do it with a custom dataset. I did however found this project 10 that loads the COCO dataset so maybe that could be of use