sjtans commited on
Commit
56b2ef3
·
verified ·
1 Parent(s): 358e1af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -308,7 +308,7 @@ def add_boxes_norm_path(filepath, boxes):
308
  print(img.shape)
309
  print(np.sum(boxes))
310
  print(boxes.shape)
311
- img[boxes] = 1
312
 
313
  fpath, fext = os.path.splitext(filepath)
314
  filepath = fpath + '_with_boxes'+'.png'
 
308
  print(img.shape)
309
  print(np.sum(boxes))
310
  print(boxes.shape)
311
+ img = np.where(boxes>0, boxes, img)
312
 
313
  fpath, fext = os.path.splitext(filepath)
314
  filepath = fpath + '_with_boxes'+'.png'