Commit
•
5f99c38
1
Parent(s):
d1e90b9
fix bug
Browse files
app.py
CHANGED
|
@@ -165,7 +165,7 @@ def inference_on_file(target_image, model, custom_test_pipeline):
|
|
| 165 |
except:
|
| 166 |
print(f'Error on image {target_image} \nContinue to next input')
|
| 167 |
|
| 168 |
-
return
|
| 169 |
|
| 170 |
def process_test_pipeline(custom_test_pipeline, bands=None):
|
| 171 |
|
|
|
|
| 165 |
except:
|
| 166 |
print(f'Error on image {target_image} \nContinue to next input')
|
| 167 |
|
| 168 |
+
return rgb1,rgb2,rgb3, result[0][0]
|
| 169 |
|
| 170 |
def process_test_pipeline(custom_test_pipeline, bands=None):
|
| 171 |
|