Samarth991 commited on
Commit
8ebe222
·
1 Parent(s): c850a01

added image detection code to display predicted bboxes

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -30,7 +30,7 @@ def draw_panoptic_segmentation(model,segmentation, segments_info):
30
 
31
  def draw_bboxes(rgb_frame,boxes,labels,color=None,line_thickness=3):
32
  rgb_frame = cv2.imread(rgb_frame)
33
- rgb_frame = cv2.cvtColor(rgb_frame,cv2.COLOR_BGR2RGB)
34
 
35
  tl = line_thickness or round(0.002 * (rgb_frame.shape[0] + rgb_frame.shape[1]) / 2) + 1 # line/font thickness
36
  rgb_frame_copy = rgb_frame.copy()
 
30
 
31
  def draw_bboxes(rgb_frame,boxes,labels,color=None,line_thickness=3):
32
  rgb_frame = cv2.imread(rgb_frame)
33
+ # rgb_frame = cv2.cvtColor(rgb_frame,cv2.COLOR_BGR2RGB)
34
 
35
  tl = line_thickness or round(0.002 * (rgb_frame.shape[0] + rgb_frame.shape[1]) / 2) + 1 # line/font thickness
36
  rgb_frame_copy = rgb_frame.copy()