Spaces:
Sleeping
Sleeping
Fix
Browse files- inference.py +5 -3
inference.py
CHANGED
@@ -65,12 +65,14 @@ def draw_boxes(frame, detections, actions, do_classify):
|
|
65 |
return frame
|
66 |
|
67 |
def draw_classes(frame, actions):
|
68 |
-
|
|
|
69 |
label_names = ['Airplane\n' + '\n'.join(actions)]
|
70 |
frame = label2rgb(
|
|
|
71 |
image=frame,
|
72 |
-
|
73 |
-
|
74 |
return frame
|
75 |
|
76 |
def FrameProcessor(detector, classifier, config: Config):
|
|
|
65 |
return frame
|
66 |
|
67 |
def draw_classes(frame, actions):
|
68 |
+
height, width, _ = frame.shape
|
69 |
+
labels = np.zeros((height, width), dtype=int)
|
70 |
label_names = ['Airplane\n' + '\n'.join(actions)]
|
71 |
frame = label2rgb(
|
72 |
+
label=labels,
|
73 |
image=frame,
|
74 |
+
label_names=label_names,
|
75 |
+
alpha=0.3)
|
76 |
return frame
|
77 |
|
78 |
def FrameProcessor(detector, classifier, config: Config):
|