Spaces:
Sleeping
Sleeping
Commit
·
8ebe222
1
Parent(s):
c850a01
added image detection code to display predicted bboxes
Browse files
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()
|