Lesson 14: Validation Predictions after Segmentation

Part 2 & Alumni (2018)
1

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!

2

Hi Ana,

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

3

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