Lesson 14: Validation Predictions after Segmentation

What variable are the predicted validation masks stored in once we’ve run the segmentation model?

I thought they were being stored in “py = to_np(learn.model(V(x)))”

But my validation set contains 200 images, and py only ever has 2 images when I try to save them. Any ideas?

Thanks in advance!

Hi Ana,

Did you ever find the answer to your question? I cannot see more than the first image prediction with that method.

Hello. Could you tell me how to save a predicted image?