remove channels last

app.py

@@ -179,12 +179,7 @@ def inference_on_file(target_image, model, custom_test_pipeline):
     time_taken=-1
     st = time.time()
     print('Running inference...')
-    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)
+    result = inference_segmentor(model, target_image, custom_test_pipeline)
     print("Output has shape: " + str(result[0].shape))
 
     ##### get metadata mask