update 目标检测结果叠加到图片中

Author: jadehh

opencv_tools/jade_opencv_process.py

@@ -390,50 +390,6 @@ def _to_color(indx):
     return b * 255, r * 255, g * 255
 
 
-# opencv显示boxes
-def CVShowBoxes(image, detectresult, num_classes=90, waitkey=-1, named_windows="result"):
-    base = int(np.ceil(pow(num_classes, 1. / 3)))
-    colors = [_to_color(x) for x in range(num_classes)]
-    if type(image) == str:
-        image = cv2.imread(image)
-    image2 = image.copy()
-    boxes = detectresult.boxes
-    for i in range(len(boxes)):
-        if boxes[i][0] <= 1 and boxes[i][1] <= 1 and boxes[i][2] <= 1 and boxes[i][3] <= 1:
-            xmin = int(boxes[i][0] * image.shape[1])
-            ymin = int(boxes[i][1] * image.shape[0])
-            xmax = int(boxes[i][2] * image.shape[1])
-            ymax = int(boxes[i][3] * image.shape[0])
-        else:
-            xmin = int(boxes[i][0])
-            ymin = int(boxes[i][1])
-            xmax = int(boxes[i][2])
-            ymax = int(boxes[i][3])
-        if boxes is not None:
-            image2 = cv2.rectangle(image2, (xmin, ymin), (xmax, ymax), GetRandomColor(), 3, 3)
-            if detectresult.label_texts is not None:
-                if detectresult.scores is not None:
-                    image2 = Add_Chinese_Label(img=image2, label=detectresult.label_texts[i] + ":" + str(
-                        int(detectresult.scores[i] * 100)),
-                                               pt1=(xmin, ymin))
-                else:
-                    image2 = Add_Chinese_Label(img=image2, label=detectresult.label_texts[i],
-                                               pt1=(xmin, ymin))
-                if detectresult.label_ids is not None:
-                    image2 = cv2.rectangle(image2, (xmin, ymin), (xmax, ymax), colors[int(detectresult.label_ids[i])],
-                                           3, 3)
-                else:
-                    image2 = cv2.rectangle(image2, (xmin, ymin), (xmax, ymax), GetRandomColor(), 3, 3)
-
-    if waitkey >= 0:
-        cv2.namedWindow(named_windows, 0)
-        # cv2.resizeWindow("result", 840, 680)
-        cv2.imshow(named_windows, image2)
-        cv2.waitKey(waitkey)
-    else:
-        return image2
-
-
 # opencv显示points
 def CVShowPoints(img_path, points, waitkey=1):
     if type(img_path) != list:
@@ -916,5 +872,5 @@ def run(self):
 if __name__ == '__main__':
     from jade import JadeLogging
     JadeLog = JadeLogging("log",Level="DEBUG")
-    videoCaptureThread = VideoCaptureBaseProcess("rtsp://admin:samples123@192.168.29.181:554/h264/ch1/main/av_stream","top",True,30,JadeLog)
+    videoCaptureThread = VideoCaptureBaseProcess("rtsp://admin:samples123@192.168.29.181:554/h264/ch1/main/av_stream","top",False,30,JadeLog)
     videoCaptureThread.start()

opencv_tools/jade_visualize.py

@@ -33,7 +33,7 @@ def get_color_map_list(num_classes):
     color_map = [color_map[i:i + 3] for i in range(0, len(color_map), 3)]
     return color_map
 
-def draw_box(im, results,show_score=True):
+def draw_box(im, results,show_score=True,font_path=None,font_size=24):
     """
     Args:
         im (PIL.Image.Image): PIL image
@@ -69,10 +69,11 @@ def draw_box(im, results,show_score=True):
             text = "{} {:.4f}".format(labels_text[i,], scores[i,])
         else:
             text = "{} ".format(labels_text[i,])
-        tw, th = draw.textsize(text)
+        font = ImageFont.truetype(get_font_path(font_path), font_size, encoding="utf-8")  # 参数1：字体文件路径，参数2：字体大小
+        tw, th = draw.textsize(text,font)
         draw.rectangle(
             [(xmin + 1, ymin - th), (xmin + tw + 1, ymin)], fill=color)
-        draw.text((xmin + 1, ymin - th), text, fill=(255, 255, 255))
+        draw.text((xmin + 1, ymin - th), text, fill=(255, 255, 255),font=font)
     return im
 
 
@@ -207,7 +208,7 @@ def draw_mask(im, np_boxes, np_masks, labels, resolution=14, threshold=0.5):
     return Image.fromarray(im.astype('uint8'))
 
 
-def visualize_box_mask(im, results, mask_resolution=14,show_score=True):
+def visualize_box_mask(im, results, mask_resolution=14,show_score=True,font_path=None,font_size=12):
     """
     Args:
         im (str/np.ndarray): path of image/np.ndarray read by cv2
@@ -233,7 +234,7 @@ def visualize_box_mask(im, results, mask_resolution=14,show_score=True):
             results["labels"],
             resolution=mask_resolution)
     if 'boxes' in results:
-        im = draw_box(im, results,show_score)
+        im = draw_box(im, results,show_score,font_path,font_size)
     if 'segm' in results:
         im = draw_segm(
             im,
@@ -248,13 +249,15 @@ def visualize_box_mask(im, results, mask_resolution=14,show_score=True):
 def visualize(image_file,
               results,
               mask_resolution=14,
-              show_score=True):
+              show_score=True,font_path=None,font_size=12):
     # visualize the predict result
     im = visualize_box_mask(
         image_file,
         results,
         mask_resolution=mask_resolution,
-        show_score=show_score)
+        show_score=show_score,
+        font_path=font_path,
+        font_size=font_size)
     return  np.array(im)