Paolo-Fraccaro commited on
Commit
66cc745
1 Parent(s): 72bf11b
Files changed (1) hide show
  1. app.py +1 -1
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[[5, 3, 2], :, :].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]
 
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]