Convert fastai ImagePoints to regular image

I am running inference with a regression model in fastai that takes in images and predicts several (x,y) coordinates:

x, y, _, _ = model.predict(image, return_x=True)

I would like to be able to plot x and y externally, i.e., I need them in regular image format.
For x, I just do torchvision.transforms.ToPILImage(mode=‘RGB’)(
However, I can’t seem to figure out how to convert the y (coordinate tensor).
I can’t use fastai’s in this case (which works fine when I’m in Jupyter Notebooks).

Thanks for any help!

I was able to get the coordinates into the right format by calling: scale_flow(FlowField(y.size,, to_unit=False).flow.flip(1) on the y coordinate tensor :+1: