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

Merge branch 'master' of https://gitea.star-rising.cn/xkrs_manan/RODY

Browse Source
This commit is contained in:
wudong
2022-11-08 10:02:48 +08:00
parent 8e7e6921f8 ee5d09adfe
commit cd01c9af86
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/yolov5/train_server.py
View File

@ -304,7 +304,7 @@ def train(hyp, opt, device, data_list,id,callbacks): # hyp is path/to/hyp.yaml
num_train_img=train_num,
train_mod_savepath=best)
# @algorithm_process_value_websocket()
@algorithm_process_value_websocket()
def report_cellback(i, num_epochs, reportAccu):
report.rate_of_progess = ((i + 1) / num_epochs) * 100
report.progress = (i + 1)
@ -470,7 +470,7 @@ def train(hyp, opt, device, data_list,id,callbacks): # hyp is path/to/hyp.yaml
print('##############',best)
for f in best:
print('##################',f)
if os.path.exists(f):
if os.path.exists(best):
strip_optimizer(f) # strip optimizers
if f is best:
LOGGER.info(f'\nValidating {f}...')