Lines Matching refs:classes

95     classes = np.argmax(box_class_probs, axis=-1)
99     classes = classes[_class_pos]
102     return boxes, classes, scores
149     boxes, classes, scores = [], [], []
154         classes.append(c)
159     classes = np.concatenate(classes)
163     for c in set(classes):
164         inds = np.where(classes == c)
166         c = classes[inds]
179     classes = np.concatenate(nclasses)
182     return boxes, classes, scores
185 def draw(image, boxes, scores, classes):  argument
195     for box, score, cl in zip(boxes, scores, classes):
307     boxes, classes, scores = yolov5_post_process(input_data)  variable
311         draw(img_1, boxes, scores, classes)