diff --git a/train.py b/train.py index 95ab9510..8df94557 100644 --- a/train.py +++ b/train.py @@ -37,7 +37,7 @@ except: hyp = {'giou': 1.582, # giou loss gain 'xy': 4.688, # xy loss gain 'wh': 0.1857, # wh loss gain - 'cls': 27.76, # cls loss gain + 'cls': 27.76, # cls loss gain (CE should be around ~1.0) 'cls_pw': 1.446, # cls BCELoss positive_weight 'obj': 21.35, # obj loss gain 'obj_pw': 3.941, # obj BCELoss positive_weight diff --git a/utils/gcp.sh b/utils/gcp.sh index 0f3051e0..a7f2c19f 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -18,7 +18,6 @@ sudo shutdown rm -rf yolov3 # Warning: remove existing git clone https://github.com/ultralytics/yolov3 # master # git clone -b test --depth 1 https://github.com/ultralytics/yolov3 test # branch -# cp -r cocoapi/PythonAPI/pycocotools yolov3 cp -r weights yolov3 && cd yolov3 # Train