Commit
·
1731256
1
Parent(s):
63d3c60
fix bug
Browse files
app.py
CHANGED
|
@@ -171,7 +171,7 @@ def inference_on_file(target_image, model, custom_test_pipeline):
|
|
| 171 |
print(f'Inference completed in {str(time_taken)} seconds')
|
| 172 |
|
| 173 |
output = (result[0][0]*18).astype(np.uint8)
|
| 174 |
-
output = np.vstack([output[None], output[None], output[None]])
|
| 175 |
|
| 176 |
return rgb1,rgb2,rgb3,output
|
| 177 |
|
|
|
|
| 171 |
print(f'Inference completed in {str(time_taken)} seconds')
|
| 172 |
|
| 173 |
output = (result[0][0]*18).astype(np.uint8)
|
| 174 |
+
output = np.vstack([output[None], output[None], output[None]]).transpose((1,2,0))
|
| 175 |
|
| 176 |
return rgb1,rgb2,rgb3,output
|
| 177 |
|