chiyoi commited on
Commit
e04c82d
1 Parent(s): a0191b1
Files changed (1) hide show
  1. 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
- label=['Airplane']
 
69
  label_names = ['Airplane\n' + '\n'.join(actions)]
70
  frame = label2rgb(
 
71
  image=frame,
72
- label=label,
73
- label_names=label_names)
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):