Spaces:
Sleeping
Sleeping
fix
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -64,7 +64,7 @@ def draw_boxes(frame, detections, actions, do_classify):
|
|
64 |
line_width=line_width)
|
65 |
return frame
|
66 |
|
67 |
-
def
|
68 |
label=['Airplane']
|
69 |
label_names = ['Airplane\n' + '\n'.join(actions)]
|
70 |
frame = label2rgb(
|
|
|
64 |
line_width=line_width)
|
65 |
return frame
|
66 |
|
67 |
+
def draw_classes(frame, actions):
|
68 |
label=['Airplane']
|
69 |
label_names = ['Airplane\n' + '\n'.join(actions)]
|
70 |
frame = label2rgb(
|