ImageClassifierData.from_csv() error


(Zane Durante) #1

(I also posted this in the Deep Learning category but got no responses so I’m going to post it here as well.)

Hi everyone.

I am trying to load bounding boxes that I saved to a csv file with pandas. I’ve manually opened the CSV file after saving it and ensured the contents are correct. However, when I call
md = ImageClassifierData.from_csv(PATH, 'imgs', BBOX_CSV, tfms=tfms, bs=bs, continuous=True) and then view the data with
next(iter(md.val_dl))
I see that a lot of my data (bounding box coordinates) have been turned to 0s. About half of the bounding boxes were turned into 0s. Does anyone have any ideas for what could be going wrong? Also, let me know if I can be more specific about anything.

Thanks!
Zane