irn icon indicating copy to clipboard operation
irn copied to clipboard

Training on own dataset

Open discretecoder opened this issue 5 years ago • 2 comments

thanks for the great work, I wish to train the network on Berkeley deep drive dataset where we have 2d bounding boxes in json files. what would be the steps? is there a data converter available? I am trying to get the dataset in pascal voc 2012 format

thanks

discretecoder avatar Oct 28 '19 09:10 discretecoder

I am trying to do the same. Have you found a way to adjust the code?

SuzannaLin avatar Sep 17 '20 08:09 SuzannaLin

What I did was convert the ground truth labels to .xml format, maintaining only the unique labels (not even including bounding box locations). Then, in the create cls_labels.py script, I changed the labels, but maintained the amount of 20 (just having empty labels). That worked for me.

SuzannaLin avatar Sep 30 '20 07:09 SuzannaLin