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

updates

This commit is contained in:
Glenn Jocher 2018-11-27 18:43:46 +01:00
parent b07ee41867
commit 5a0575af3a
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -15,7 +15,7 @@ parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each i
parser.add_argument('-resume', default=False, help='resume training flag')
parser.add_argument('-batch_report', default=False, help='report TP, FP, FN, P and R per batch (slower)')
parser.add_argument('-optimizer', default='SGD', help='optimizer')
parser.add_argument('-freeze_darknet53', default=False, help='freeze darknet53.conv.74 layers for first epoch')
parser.add_argument('-freeze_darknet53', default=True, help='freeze darknet53.conv.74 layers for first epoch')
parser.add_argument('-var', type=float, default=0, help='optional test variable')
opt = parser.parse_args()
print(opt)

11
utils/gcp.sh
View File

@ -30,3 +30,14 @@ gsutil cp yolov3/weights/latest.pt gs://ultralytics
# Copy latest.pt from bucket
gsutil cp gs://ultralytics/latest.pt yolov3/weights/latest.pt
# Testing
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3
python3 train.py -epochs 1 -var 1
python3 train.py -epochs 1 -var 2
python3 train.py -epochs 1 -var 4
python3 train.py -epochs 1 -var 8
python3 train.py -epochs 1 -var 16
python3 train.py -epochs 1 -var 32
python3 train.py -epochs 1 -var 64
sudo shutdown