Browse Source

gengxin

master
bai 3 years ago
parent
commit
3116935302
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      yolo_mindspore/yolo_to_coco.py

+ 2
- 2
yolo_mindspore/yolo_to_coco.py View File

@@ -5,12 +5,12 @@ import time
from PIL import Image from PIL import Image
import csv import csv


coco_format_save_path = './coco' # 要生成的标准coco格式标签所在文件夹
coco_format_save_path = './coco' # 要生成的coco格式标签所在文件夹
yolo_format_classes_path = 'annotations.csv' # 类别文件,用csv文件表示,一行一个类 yolo_format_classes_path = 'annotations.csv' # 类别文件,用csv文件表示,一行一个类
yolo_format_annotation_path = './dataset/mask/labels/val' # yolo格式标签所在文件夹 yolo_format_annotation_path = './dataset/mask/labels/val' # yolo格式标签所在文件夹
img_pathDir = './dataset/mask/images/val' # 图片所在文件夹 img_pathDir = './dataset/mask/images/val' # 图片所在文件夹


# 类别设置
# 类别设置
categories = [] categories = []
class_names = ['with_mask', 'without_mask'] class_names = ['with_mask', 'without_mask']
for label in class_names: for label in class_names:


Loading…
Cancel
Save