Commit
•
5eb1e83
1
Parent(s):
77e0567
fix bug
Browse files
app.py
CHANGED
|
@@ -158,7 +158,7 @@ def inference_on_file(target_image, model, custom_test_pipeline):
|
|
| 158 |
time_taken = np.round(et - st, 1)
|
| 159 |
print(f'Inference completed in {str(time_taken)} seconds')
|
| 160 |
|
| 161 |
-
return rgb/10000, result[0][0]*255
|
| 162 |
|
| 163 |
def process_test_pipeline(custom_test_pipeline, bands=None):
|
| 164 |
|
|
|
|
| 158 |
time_taken = np.round(et - st, 1)
|
| 159 |
print(f'Inference completed in {str(time_taken)} seconds')
|
| 160 |
|
| 161 |
+
return (rgb/10000*255).astype(np.uint8), result[0][0]*255
|
| 162 |
|
| 163 |
def process_test_pipeline(custom_test_pipeline, bands=None):
|
| 164 |
|