调试

app/controller/AlgorithmController.py

@@ -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/",

app/yolov5/train_server.py

@@ -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')