xkrs_manan/RODY
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

调试

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
app/controller/AlgorithmController.py
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/",