调试

2022-11-08 08:59:45 +08:00
parent a78ef10b8a
commit 9e5134628e
3 changed files with 2 additions and 8 deletions
--- a/app/controller/AlgorithmController.py
+++ b/app/controller/AlgorithmController.py
@ -249,7 +249,6 @@ 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('CLASS_NAMES').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'  # 模型命名加上图像参数