| | |
| | | cv2.putText(img, label, (x - 10, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2) |
| | | |
| | | def draw_detections(self, frame, class_ids, scores, boxes): |
| | | det_img = frame.copy() |
| | | result_boxes = cv2.dnn.NMSBoxes(boxes, scores, 0.25, 0.45, 0.5) |
| | | |
| | | for i in range(len(result_boxes)): |
| | | index = result_boxes[i] |
| | | box = boxes[index] |
| | | self.draw_bounding_box(frame, class_ids[index], scores[index], round(box[0] * self.scale), round(box[1] * self.scale), |
| | | self.draw_bounding_box(det_img, class_ids[index], scores[index], round(box[0] * self.scale), round(box[1] * self.scale), |
| | | round((box[0] + box[2]) * self.scale), round((box[1] + box[3]) * self.scale)) |
| | | return frame |
| | | return det_img |
| | | |
| | | |
| | | |