[[enter image description here](https://i.sstatic.net/VhxrAQth.png)](https://i.sstatic.net/4DWyqfLj.png)

I am currently using new images to train a pre-trained model (https://doi.org/10.1104/pp.19.00728) to segment plants, but I can only get a segmentation map with a blue background. I want to get a map with a red background because the blue is the target area I want to segment, but I don’t know where the problem lies.

At present, I am considering the reason that the number of training epochs is insufficient, so I increased it to 1000, but the result did not change much. I also tried to modify the code, but it was unsuccessful. The preprocessing of the pictures was also carried out according to the tutorial in the paper.