diff --git a/app/controller/AlgorithmController.py b/app/controller/AlgorithmController.py index 6ddc2fe..ac78e7f 100644 --- a/app/controller/AlgorithmController.py +++ b/app/controller/AlgorithmController.py @@ -249,6 +249,7 @@ def train_R0DY(params_str, id): 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) + print(data_list) weights = params.get('resumeModPath').value # 初始化模型绝对路径 img_size = params.get('img_size').value savemodel = os.path.splitext(params.get('saveModDir').value)[0] + '_' + str(img_size) + '.pt' # 模型命名加上图像参数