From b450db18ae3530d4d5fda8193f838a381880bde4 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 16 Aug 2019 11:08:07 +0200 Subject: [PATCH] updates --- train.py | 2 +- utils/gcp.sh | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) 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