|
|
|
@@ -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: |
|
|
|
|