PointsItemList -- Center is off

(Gokul) #1

Hi Guys,

I am trying to solve an Image regression problem and using the code from lesson3 (head-pose-detection).

The problem is when I do show_batch on the databunch, the y-label seems to be off.
In the following image, the first part(blue dot) is the actual label, but the dataloader is plotting(red dot) it away from its intended location.

Would be thankful for any pointers on fixing this.

Thanks,
Gokul

0 Likes

#2

Your .transform has tfm_y=False, meaning that the resizing transformation is not applied to the ImagePoint. Have you tried changing it to True?

0 Likes

(Michael Benedict) #3

fastai expects points to have the y coordinate first, so try reversing the order of the points you return from get_labels().

1 Like

(Gokul) #4

Bulls eye, Michael! Thanks :slight_smile:

0 Likes