Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-12-21 19:47:49 -08:00
parent 66fe3db8fb
commit b7a53957b3
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -446,7 +446,7 @@ if __name__ == '__main__':
mixed_precision = False mixed_precision = False
# scale hyp['obj'] by img_size (evolved at 416) # scale hyp['obj'] by img_size (evolved at 416)
hyp['obj'] *= opt.img_size / 416. hyp['obj'] *= opt.img_size / 320.
tb_writer = None tb_writer = None
if not opt.evolve: # Train normally if not opt.evolve: # Train normally

10
utils/evolve.sh Normal file
View File

@ -0,0 +1,10 @@
#for i in 1 2 3 4 5 6 7
#do
# t=ultralytics/yolov3:v139 && sudo docker pull $t && sudo nvidia-docker run --d --ipc=host -v "$(pwd)"/coco:/usr/src/coco $t utils/evolve.sh $i
# sleep 30
# done
while true
do
python3 train.py --data coco2014.data --img-size 320 --epochs 27 --batch-size 64 --accumulate 1 --evolve --weights '' --pre --bucket yolov4/320_coco2014_27e --device $1
done