labelme标注后的数据去除单点和双点的标签

单点和双点的错误标签转化不成多边形，导致不能转化成实例分割的训练集所以在训练前需要去除掉

点击查看代码

import cv2
import numpy as np
import json
import os
# 去除一两个点的小polygon
def remove_specific_labels(json_file):
    # 读取JSON文件
    with open(json_file, 'rb+') as f:
        data = json.load(f)
    if"shapes" in data:
        new_shapes=[]
        for shape in data['shapes']:
            points = np.array(shape['points'], np.int32)
            if(len(points)<3):
                print("单点polygon",points)
            else:
                new_shapes.append(shape)
        data["shapes"]=new_shapes
        # data_small["shapes"]=new_shapes_small
    with open(json_file,'w',encoding='utf-8')as f:
        json.dump(data,f,ensure_ascii=False,indent=4)


def traverse_jsons_folder(folder_path):
    image_files = []
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            if file.endswith(".json"):
                print(file)
                remove_specific_labels(os.path.join(root,file))

if __name__ == '__main__':
    # remove_specific_labels(r'D:\pic\see\0906\src\add24.json')
    traverse_jsons_folder('/home/administrator/gzj/pic/primary_tiao/src_copy/')