I am using the pytorch resnet for the first time model = models.segmentation.fcn_resnet50(pretrained=True) to classify satellite images of land. My goal is to get the percentage of the land covered by forests. For input I have satellite images and there is a mask associated with them which is color coded (forests,farmland,urban area, water…).
I made my own Data Structure in pytorch and then call the model but somewhere it’s going wrong. I think it’s because I’m not inputing the color coding but I’m not sure.
Here is what it should look like:
Expected
And here is what happens:
What I get