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 |
|