xkrs_manan/RODY
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

共享传参

Browse Source
This commit is contained in:
JIAKUNHAO 2022-11-25 17:07:45 +08:00
parent 9a4b11b054
commit 00dc9b02eb
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/controller/AlgorithmController.py
View File

@ -301,7 +301,7 @@ def error_return(id: str, data):
# 启动训练 # 启动训练
@start_train_algorithm() @start_train_algorithm()
def train_R0DY(params_str, id): def train_R0DY(params_str, id, getsomething):
print('**********************************') print('**********************************')
print(params_str) print(params_str)
print('**********************************') print('**********************************')
@ -317,7 +317,7 @@ def train_R0DY(params_str, id):
batch_size = params.get('batch_size').value batch_size = params.get('batch_size').value
device = params.get('device').value device = params.get('device').value
#try: #try:
train_start(weights, savemodel, epoches, img_size, batch_size, device, data_list, id) train_start(weights, savemodel, epoches, img_size, batch_size, device, data_list, id, getsomething)
print("train down!") print("train down!")
# except Exception as e: # except Exception as e:
# print(repr(e)) # print(repr(e))

12
app/yolov5/train_server.py
View File

@ -89,7 +89,7 @@ def yaml_rewrite(file='data.yaml',data_list=[]):
with open(file, 'w') as f: with open(file, 'w') as f:
yaml.safe_dump(coco_dict, f, sort_keys=False) yaml.safe_dump(coco_dict, f, sort_keys=False)
def train(hyp, opt, device, data_list,id,callbacks): # hyp is path/to/hyp.yaml or hyp dictionary def train(hyp, opt, device, data_list,id,getsomething,callbacks): # hyp is path/to/hyp.yaml or hyp dictionary
#将数据路径写到yaml文件中 #将数据路径写到yaml文件中
#data_list = file_tool.get_file(proj_no=pro) #data_list = file_tool.get_file(proj_no=pro)
# print(data_list) # print(data_list)
@ -574,7 +574,7 @@ def parse_opt(weights,savemodel,epoches,img_size,batch_size,device,known=False):
return parser.parse_known_args()[0] if known else parser.parse_args() return parser.parse_known_args()[0] if known else parser.parse_args()
def main(opt,data_list,id,callbacks=Callbacks()): def main(opt,data_list,id,getsomething,callbacks=Callbacks()):
# Checks # Checks
if RANK in {-1, 0}: if RANK in {-1, 0}:
print_args(vars(opt)) print_args(vars(opt))
@ -624,7 +624,7 @@ def main(opt,data_list,id,callbacks=Callbacks()):
# Train # Train
if not opt.evolve: if not opt.evolve:
train(opt.hyp, opt, device, data_list,id,callbacks) train(opt.hyp, opt, device, data_list,id,getsomething,callbacks)
# Evolve hyperparameters (optional) # Evolve hyperparameters (optional)
else: else:
@ -705,7 +705,7 @@ def main(opt,data_list,id,callbacks=Callbacks()):
hyp[k] = round(hyp[k], 5) # significant digits hyp[k] = round(hyp[k], 5) # significant digits
# Train mutation # Train mutation
results = train(hyp.copy(), opt, device, callbacks) results = train(hyp.copy(), opt, device, getsomething=getsomething,callbacks=callbacks)
callbacks = Callbacks() callbacks = Callbacks()
# Write mutation results # Write mutation results
print_mutation(results, hyp.copy(), save_dir, opt.bucket) print_mutation(results, hyp.copy(), save_dir, opt.bucket)
@ -725,9 +725,9 @@ def run(**kwargs):
main(opt) main(opt)
return opt return opt
def train_start(weights,savemodel,epoches,img_size,batch_size,device,data_list,id): def train_start(weights,savemodel,epoches,img_size,batch_size,device,data_list,id,getsomething):
opt = parse_opt(weights,savemodel,epoches,img_size,batch_size,device) opt = parse_opt(weights,savemodel,epoches,img_size,batch_size,device)
main(opt,data_list,id) main(opt,data_list,id,getsomething)
if __name__ == "__main__": if __name__ == "__main__":