Test with pretrained model

[SarahwXU]

Hello, May I ask how to test other images with your pretrained model (158 layers) on Inria Aerial Image Labeling Dataset ?

When I am running the 'run_prediction.py', I get the error: ValueError: Shapes must be equal rank, but are 4 and 1