This commit is contained in:
552068321@qq.com
2022-11-08 09:59:54 +08:00
parent 317a71ed5f
commit ee5d09adfe
4 changed files with 9 additions and 9 deletions

View File

@ -242,7 +242,7 @@ from app import file_tool
# 启动训练
@start_train_algorithm()
#@start_train_algorithm()
def train_R0DY(params_str, id):
from app.yolov5.train_server import train_start
params = TrainParams()
@ -314,7 +314,7 @@ def train_R0DY(params_str, id):
# zip_outputPath = os.path.join(exp_outputPath, "inference_model.zip")
@obtain_train_param()
#@obtain_train_param()
def returnTrainParams():
# nvmlInit()
# gpuDeviceCount = nvmlDeviceGetCount() # 获取Nvidia GPU块数
@ -338,7 +338,7 @@ def returnTrainParams():
{"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/",
{"index": 8, "name": "DatasetDir", "value": "E:/aicheck/data_set/11442136178662604800/ori",
"description": '数据集路径',
"default": "./app/maskrcnn/datasets/test", "type": "S", 'show': False} # ORI_PATH
]