Paolo-Fraccaro
commited on
Commit
•
66cc745
1
Parent(s):
72bf11b
fix bug
Browse files
app.py
CHANGED
@@ -140,7 +140,7 @@ def inference_on_file(target_image, model, custom_test_pipeline):
|
|
140 |
##### get metadata mask
|
141 |
mask = open_tiff(target_image)
|
142 |
# rgb = mask[[2, 1, 0], :, :].transpose((1,2,0))
|
143 |
-
rgb = mask[[
|
144 |
meta = get_meta(target_image)
|
145 |
mask = np.where(mask == meta['nodata'], 1, 0)
|
146 |
mask = np.max(mask, axis=0)[None]
|
|
|
140 |
##### get metadata mask
|
141 |
mask = open_tiff(target_image)
|
142 |
# rgb = mask[[2, 1, 0], :, :].transpose((1,2,0))
|
143 |
+
rgb = mask[[3, 2, 1], :, :].transpose((1,2,0))
|
144 |
meta = get_meta(target_image)
|
145 |
mask = np.where(mask == meta['nodata'], 1, 0)
|
146 |
mask = np.max(mask, axis=0)[None]
|