调试
This commit is contained in:
@ -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
|
||||
]
|
||||
|
Reference in New Issue
Block a user