carlosgomes98 commited on
Commit
ef84153
1 Parent(s): 842ee21

show error

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -142,12 +142,7 @@ def inference_on_file(target_image, model, custom_test_pipeline):
142
 
143
  st = time.time()
144
  print('Running inference...')
145
- try:
146
- result = inference_segmentor(model, target_image, custom_test_pipeline)
147
- except:
148
- print('Error: Try different channels order.')
149
- model.cfg.data.test.pipeline[0]['channels_last'] = True
150
- result = inference_segmentor(model, target_image, custom_test_pipeline)
151
 
152
  print("Output has shape: " + str(result[0].shape))
153
 
 
142
 
143
  st = time.time()
144
  print('Running inference...')
145
+ result = inference_segmentor(model, target_image, custom_test_pipeline)
 
 
 
 
 
146
 
147
  print("Output has shape: " + str(result[0].shape))
148