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

2022-11-23 19:53:10 +08:00
parent 1aecd2ab27 2b522c32cf
commit 728523b2e5
3 changed files with 6 additions and 9 deletions
--- a/app/yolov5/train_server.py
+++ b/app/yolov5/train_server.py
@ -326,6 +326,8 @@ def train(hyp, opt, device, data_list,id,callbacks):  # hyp is path/to/hyp.yaml
        report.progress = (i + 1)
        report.end_time = datetime.now()
        report.precision[0].value.append(reportAccu)
+        if (i+1) == num_epochs:
+            report.isfinish = 2
        return report.dict()
    ###################结束#######################
    for epoch in range(start_epoch, epochs):  # epoch ------------------------------------------------------------------