This commit is contained in:
552068321@qq.com 2022-11-07 20:42:17 +08:00
parent 10a84a524e
commit 2af24e2f03
2 changed files with 3 additions and 2 deletions

View File

@ -248,7 +248,7 @@ def train_R0DY(params_str, id):
params = TrainParams()
params.read_from_str(params_str)
print(params.get('device').default)
data_list = file_tool.get_file(ori_path=params.get('DatasetDir').value, type_list=params.get('classname').value)
data_list = file_tool.get_file(ori_path=params.get('DatasetDir').value, type_list=params.get('CLASS_NAMES').value)
print('---------------------',data_list)
weights = params.get('resumeModPath').value # 初始化模型绝对路径
img_size = params.get('img_size').value
@ -336,7 +336,7 @@ def returnTrainParams():
'show': False},
{"index": 6, "name": "resumeMod", "value": '', "description": '继续训练模型', "default": '', "type": "E", "items": '',
'show': True},
{"index": 7, "name": "classname", "value": ['hole', '456'], "description": '类别名称', "default": '', "type": "L",
{"index": 7, "name": "CLASS_NAMES", "value": ['hole', '456'], "description": '类别名称', "default": '', "type": "L",
"items": '',
'show': False},
{"index": 8, "name": "DatasetDir", "value": "E:/aicheck/data_set/11442136178662604800/ori/",

View File

@ -501,6 +501,7 @@ def parse_opt(weights,savemodel,epoches,img_size,batch_size,device,known=False):
if weights == None:
weights = ROOT /'yolov5s.pt'
print('ROOT###############',ROOT)
parser = argparse.ArgumentParser()
parser.add_argument('--weights', type=str, default=weights, help='initial weights path')
parser.add_argument('--savemodel', type=str, default=savemodel, help='initial weights path')