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

0 Likes

#2

Did you figure this out? I am seeing something similar and I am trying to figure this out.

Thanks!

-Dimitris

0 Likes

(Zane Durante) #3

I didn’t necessarily figure this out, but I could work around the issue by removing the data transformations I was using. (Basically I didn’t use the normalize and the later denorm steps).

I also tried messing around with feeding the data in as numpy arrays (if removing the transforms themselves doesn’t fix it) and that worked as well.

Let me know if this helps,
Zane

0 Likes