调试

2022-11-07 20:42:17 +08:00
parent 10a84a524e
commit 2af24e2f03
2 changed files with 3 additions and 2 deletions
--- a/app/yolov5/train_server.py
+++ b/app/yolov5/train_server.py
@ -501,6 +501,7 @@ def parse_opt(weights,savemodel,epoches,img_size,batch_size,device,known=False):

    if weights == None:
        weights = ROOT /'yolov5s.pt'
+    print('ROOT###############',ROOT)
    parser = argparse.ArgumentParser()
    parser.add_argument('--weights', type=str, default=weights, help='initial weights path')
    parser.add_argument('--savemodel', type=str, default=savemodel, help='initial weights path')