3.数据处理把labelme标注的关键点转成coco格式.docxVIP

3.数据处理-把labelme标注的关键点转成coco格式 # *_* : coding: utf-8 *_* ##################### #cao datasets process for object detection project. for convert customer dataset format to coco data format, import traceback import argparse import datetime import json import cv2 import os __CLASS__ = [__background__, lpr] # class dictionary, background must be in first index. def argparser(): parser = argparse.ArgumentParser(define argument parser for pycococreator!) #图?和标签的路径 parser.add_argument(-r, --project_path, default=/mnt//, help=path of root directory) parser.add_argument(-l, --label_path, default=/mn/labels, help=path of root directory) #图?所在的?件夹，可以?持多个?件夹 parser.add_argument(-p, --phase_folder, default=[ceme], help=datasets path of [train, val, test]) #关键点个数 parser.add_argument(-joint, --pointNum, default=33,help=point numbers) # 是否开启关键点读取， parser.add_argument(-po, --have_points, default=True, help=if have points we will deal it!) # 是否开启把点画在图?上验证读取， parser.add_argument(-check, --have_check, default=True, help=if you want to check all points!) return parser.parse_args() def MainProcessing(args): main process source code. annotations = {} # annotations dictionary, which will dump to json format file. project_path = ject_path phase_folder = args.phase_folder # coco annotations info. annotations[info] = { description: customer dataset format convert to COCO format, url: , version: 1.0, year: 2020, contributor: andy.wei, date_created: 2020/08/24 } # coco annotations licenses. annotations[licenses] = [{ url: https:// /licenses/LICENSE-2.0.html, id: 1, name: Apache License 2.0 }] # coco annotations categories. annotations[categories] = [] for cls, clsname in enumerate(__CLASS__): if clsname == __background__: continue annotations[categories].append( { supercategory: object, id: cls, name: clsname } ) #保存??的标签 for catdict in annotations[categories]: if lpr == catdict[name] and args.have_points: catdict[keypoints] = [LT1, LT2, LT3, LT4, LT5,LB1, LB2, LB3, LB4, LB5, RT1, RT2, RT3,