Merge pull request #21 from Epoxid2/master

Added check for multi-layer tiffs

starnet_v1_TF2.py

@@ -306,14 +306,19 @@ def save_model(self, weights_filename, history_filename = None):
 
     def transform(self, in_name, out_name):
         data = tiff.imread(in_name)
+        if len(data.shape) > 3:
+            layer = input("Tiff has %d layers, please enter layer to process: "%data.shape[0])
+            layer = int(layer)
+            data=data[layer]
+
         input_dtype = data.dtype
         if input_dtype == 'uint16':
             image = (data / 255.0 / 255.0).astype('float32')
         elif input_dtype == 'uint8':
             image = (data / 255.0).astype('float32')
         else:
             raise ValueError('Unknown image dtype:', data.dtype)
-
+            
         if self.mode == 'Greyscale' and len(image.shape) == 3:
             raise ValueError('You loaded Greyscale model, but the image is RGB!')