Solved by vmirly1 in post #2 
              
              
                You have to make sure that the distribution of the test data are similar to the training data. 


First, make sure that the sequence of transformations applied to train and test data result in the same range of values. For example, it can be that in the training data, the background is denoted as 1 …