fix channels order

app.py

@@ -134,7 +134,12 @@ def inference_on_file(target_image, model, custom_test_pipeline):
     
     st = time.time()
     print('Running inference...')
-    result = inference_segmentor(model, target_image, custom_test_pipeline)
+    try:
+        result = inference_segmentor(model, target_image, custom_test_pipeline)
+    except:
+        print('Error: Try different channels order.')
+        model.cfg.data.test.pipeline[0]['channels_last'] = True
+        result = inference_segmentor(model, target_image, custom_test_pipeline)
     print("Output has shape: " + str(result[0].shape))
 
     # prep outputs