From 735c7c514db3fe8f574a7709cf84edab210c802d Mon Sep 17 00:00:00 2001 From: "552068321@qq.com" Date: Mon, 7 Nov 2022 20:55:55 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/yolov5/train_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/yolov5/train_server.py b/app/yolov5/train_server.py index 79e9292..72348a9 100644 --- a/app/yolov5/train_server.py +++ b/app/yolov5/train_server.py @@ -54,7 +54,6 @@ from app.yolov5.utils.general import (LOGGER, check_amp, check_dataset, check_fi init_seeds, intersect_dicts, labels_to_class_weights, labels_to_image_weights, methods, one_cycle, print_args, print_mutation, strip_optimizer, yaml_save) from app.yolov5.utils.loggers import Loggers -#from yolov5.utils.loggers.wandb.wandb_utils import check_wandb_resume from app.yolov5.utils.loss import ComputeLoss from app.yolov5.utils.metrics import fitness from app.yolov5.utils.plots import plot_evolve @@ -78,7 +77,7 @@ def yaml_rewrite(file='data.yaml',data_list=[]): f.close() classes = class_dict["classes"] nc, names = (len(classes), classes) - #print(names) + print("names##############",names) coco_dict['train'] = data_list[0] coco_dict['val'] = data_list[1] label_name = {} @@ -86,6 +85,7 @@ def yaml_rewrite(file='data.yaml',data_list=[]): label_name[i] = names[i] #print(label_name) coco_dict['names'] = label_name + print('coco_names################:',coco_dict['names']) with open(file, 'w') as f: yaml.safe_dump(coco_dict, f, sort_keys=False)