ai.detect: anchor region box

opencv-yolo/web/detect_video.py

@@ -23,7 +23,6 @@ def get_args():
     args = parser.parse_args()
     return args
 
-
 def test(opt):
     tsEpoch = datetime.datetime.utcfromtimestamp(0)
     model = torch.load(opt.pretrained_model, map_location='cpu').module
@@ -128,8 +127,10 @@ def test(opt):
             xmin, ymin, xmax, ymax = boxes[box_id, :]
             color = colors[pred_label]
             color = (255, 0, 0)
+            colorG = (0, 255, 0)
             font = cv2.FONT_HERSHEY_PLAIN
             cv2.rectangle(output_image, (xmin, ymin), (xmax, ymax), color, 1)
+            cv2.rectangle(output_image, (px1, py1), (px2, py2), colorG, 1)
             text_size = cv2.getTextSize(
                 COCO_CLASSES[pred_label] + ' : %.2f' % pred_prob, font, 1, 1)[0]
             cv2.rectangle(output_image, (xmin, ymin), (xmin +