新增图片添加中文字符串支持

Author: pi-pi

README.md

@@ -32,7 +32,7 @@ Code segment are often used in deep learning algorithms(pytorch/numpy)
 
 ### 2 img_show
 
-​    opencv写的图片显示功能，支持单张图、多张图、多张图合并、多张图自定义行列数显示功能；支持bbox显示
+​    opencv写的图片显示功能，支持单张图、多张图、多张图合并、多张图自定义行列数显示功能；支持bbox显示；支持中文显示
 
 ### 3 pr_roc
 

docs/detail.md

@@ -36,7 +36,18 @@ ImageHelper.show_bbox(img, bbox_list, color=(255, 0, 0), font_scale=0.2, thickne
 ImageHelper.show_bbox(img, bbox_list, color=(255, 0, 0), font_scale=0.2, thickness=1,is_without_mask=True)
 ```
 
+支持中文
+
+```python
+img = ImageHelper.cv2ImgAddText(img, '这是一个测试中文代码', 10, 10, textSize=8, font='./NotoSansCJK-Bold')
+```
+
+需要注意的是需要字体库支持，我采用的是NotoSansCJK-Bold，后缀是.ttc，ubuntu系统下有，可以查找出来。
+
+
+
 ## 3 pr_roc
+
 [c3_pr_roc](../library/c3_pr_roc.py) 
 
 ​    分类问题中，对于类别不平衡问题，采用acc指标是不够的，pr曲线和roc曲线绘制来评估模型性能比较关键。

library/c2_img_show.py

@@ -200,6 +200,16 @@ def convert_gray_to_bgr(img):
         bgr_img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
         return bgr_img
 
+    @staticmethod
+    def cv2ImgAddText(img, text, left, top, textColor=(0, 255, 0), textSize=20, font='./NotoSansCJK-Bold.ttc'):
+        from PIL import Image, ImageDraw, ImageFont
+        if isinstance(img, np.ndarray):  # 判断是否OpenCV图片类型
+            img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
+        draw = ImageDraw.Draw(img)
+        fontText = ImageFont.truetype(font, textSize, encoding="utf-8")
+        draw.text((left, top), text, textColor, font=fontText)
+        return cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)
+
     @staticmethod
     def show_img(imgs, window_names=None, wait_time_ms=0, is_merge=False, row_col_num=(1, -1)):
         """
@@ -333,4 +343,4 @@ def show_bbox(image, bboxs_list, color=None,
                             font, font_scale, (255, 255, 255), thickness=thickness, lineType=cv2.LINE_AA)
         if is_show:
             ImageHelper.show_img(image_copy, names, wait_time_ms)
-        return image_copy
+        return image_copy

tools/c2_img_show.py

@@ -29,6 +29,13 @@ def demo_img_show():
     ImageHelper.show_img([img, img2, img3, img4, img5], is_merge=True, row_col_num=(2, 3))  # 合并显示，指定数目
 
 
+    # 显示中文,需要字体库
+    # 输入图片默认是opencv读出来的bgr格式
+    img = ImageHelper.cv2ImgAddText(img, '这是一个测试中文代码', 10, 10, textSize=8, font='./NotoSansCJK-Bold')
+    ImageHelper.show_img(img)
+
+
+
 def demo_bbox_show():
     input_shape = [400, 400, 3]
     img = np.zeros(input_shape, np.uint8)