carlosgomes98
commited on
Commit
•
ef84153
1
Parent(s):
842ee21
show error
Browse files
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 |
-
|
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 |
|