Change shape of tiff image

SingleCellDataExtraction.py

@@ -92,7 +92,7 @@ def PrepareData(mask,image,channel_names):
             image_loaded = skimage.io.imread(image,plugin='tifffile')
             #print('TIF(F) found')
             # Remove extra axis
-            image_loaded = image_loaded.reshape((image_loaded.shape[0],image_loaded.shape[3],image_loaded.shape[4]))
+            image_loaded = image_loaded.reshape((image_loaded.shape[1],image_loaded.shape[3],image_loaded.shape[4]))
 
     #Check to see if image is hdf5
     elif image_path.suffix == '.h5' or image_path.suffix == '.hdf5':
@@ -183,4 +183,4 @@ def MultiExtractSingleCells(mask,image,channel_names,output):
 
     im_full_name = os.path.basename(image)
     im_name = im_full_name.split('.')[0]
-    print("Finished "+str(im_name))
+    print("Finished "+str(im_name))