调试

app/yolov5/data/coco128.yaml

@@ -3,5 +3,5 @@ train: E:/aicheck/data_set/11442136178662604800/trained/images/train/
 val: E:/aicheck/data_set/11442136178662604800/trained/images/val/
 test: null
 names:
-  0: logo
-  1: 3C
+  0: hole
+  1: '456'

app/yolov5/train_server.py

@@ -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}...')