updates
This commit is contained in:
parent
dbb2cbe0d5
commit
b450db18ae
2
train.py
2
train.py
|
@ -37,7 +37,7 @@ except:
|
||||||
hyp = {'giou': 1.582, # giou loss gain
|
hyp = {'giou': 1.582, # giou loss gain
|
||||||
'xy': 4.688, # xy loss gain
|
'xy': 4.688, # xy loss gain
|
||||||
'wh': 0.1857, # wh 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
|
'cls_pw': 1.446, # cls BCELoss positive_weight
|
||||||
'obj': 21.35, # obj loss gain
|
'obj': 21.35, # obj loss gain
|
||||||
'obj_pw': 3.941, # obj BCELoss positive_weight
|
'obj_pw': 3.941, # obj BCELoss positive_weight
|
||||||
|
|
|
@ -18,7 +18,6 @@ sudo shutdown
|
||||||
rm -rf yolov3 # Warning: remove existing
|
rm -rf yolov3 # Warning: remove existing
|
||||||
git clone https://github.com/ultralytics/yolov3 # master
|
git clone https://github.com/ultralytics/yolov3 # master
|
||||||
# git clone -b test --depth 1 https://github.com/ultralytics/yolov3 test # branch
|
# 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
|
cp -r weights yolov3 && cd yolov3
|
||||||
|
|
||||||
# Train
|
# Train
|
||||||
|
|
Loading…
Reference in New Issue