souranil3d commited on
Commit
5eb3208
·
1 Parent(s): 74fb45b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -30,6 +30,7 @@ def predict(
30
  predictor = DefaultPredictor(cfg)
31
  im = cv2.imread(img.name)
32
  output = predictor(im)
 
33
  img = add_bboxes(im, output['instances'].pred_boxes, scores=output['instances'].scores)
34
  return img
35
 
 
30
  predictor = DefaultPredictor(cfg)
31
  im = cv2.imread(img.name)
32
  output = predictor(im)
33
+ output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
34
  img = add_bboxes(im, output['instances'].pred_boxes, scores=output['instances'].scores)
35
  return img
36