Mask to Polygons mask转Polygons并展示

import base64
import copy
import glob
import json
import sys
import time
import cv2 as cv
from paddle.vision import transforms
import os
import numpy as np
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont


label_dict = {
    
    '十分大方': 1,  #
    '按时发生': 2,  #
    '暗示法规规定': 3,  #
    '哈根达斯公司第': 4,  #
    '粉色的': 5,  #
    'gas': 6,  #
    '保阿斯蒂芬棚': 7  #
}
class Mask2Polygons():
    def test_mask_2_polygons_show_by_split_mask(self, mask, img_path, label_dict):
        """按类拆分mask,通过获取mask的Contours展示标签"""
        lable_k = list(label_dict.keys())
        label_v = list(label_dict.values())
        if type(img_path) is str:
            img_np = cv.imread(img_path)
        # 以颜色快展示mask,
        color_mask = np.zeros_like(img_np)
        for v_i, v in enumerate(label_v):
            # 为类赋值颜色
            label_index = np.argwhere(mask == v)
            if len(label_index) != 0:
                color_mas
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38