From 311693530296b6a7387272bb344dc1380f39d17d Mon Sep 17 00:00:00 2001 From: bai <996950537@qq.com> Date: Tue, 18 Oct 2022 19:51:04 +0800 Subject: [PATCH] gengxin --- yolo_mindspore/yolo_to_coco.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/yolo_mindspore/yolo_to_coco.py b/yolo_mindspore/yolo_to_coco.py index c601d62..501dd81 100644 --- a/yolo_mindspore/yolo_to_coco.py +++ b/yolo_mindspore/yolo_to_coco.py @@ -5,12 +5,12 @@ import time from PIL import Image import csv -coco_format_save_path = './coco' # 要生成的标准coco格式标签所在文件夹 +coco_format_save_path = './coco' # 要生成的coco格式标签所在文件夹 yolo_format_classes_path = 'annotations.csv' # 类别文件,用csv文件表示,一行一个类 yolo_format_annotation_path = './dataset/mask/labels/val' # yolo格式标签所在文件夹 img_pathDir = './dataset/mask/images/val' # 图片所在文件夹 -# 类别设置 +# 类别设置 categories = [] class_names = ['with_mask', 'without_mask'] for label in class_names: