car-detection-bayes/our_scripts/config.yml

59 lines
1.7 KiB
YAML
Raw Normal View History

2020-06-28 12:37:21 +00:00
train:
2020-07-02 16:10:02 +00:00
epochs: 1200
batch-size: 3
2020-06-30 19:51:10 +00:00
cfg: ./cfg/yolov3-spp-19cls.cfg
data: ./data/widok_01_19.data
2020-06-28 12:37:21 +00:00
multi-scale: false
2020-07-02 16:10:02 +00:00
img-size: '512 1920'
rect: true
resume: false
2020-06-28 12:37:21 +00:00
nosave: false
notest: false
evolve: false
bucket:
cache-images: false
2020-07-02 16:10:02 +00:00
weights: /home/tomekb/yolov3/weights/yolov3-spp-ultralytics.pt
2020-06-28 12:37:21 +00:00
device: 1
adam: true
single-cls: false
2020-07-02 16:10:02 +00:00
save-every-nth-epoch: 50
2020-06-28 12:37:21 +00:00
# inne hiperparametry
other-hyps:
giou: 3.53 # giou loss gain
cls: 37.4 # cls loss gain
cls_pw: 1.0 # cls BCELoss positive_weight
obj: 64.3 # obj loss gain (*=img_size/320 if img_size != 320)
obj_pw: 1.0 # obj BCELoss positive_weight
iou_t: 0.20 # iou training threshold
lr0: 0.01 # initial learning rate (SGD=5E-3 Adam=5E-4)
lrf: 0.0005 # final learning rate (with cos scheduler)
momentum: 0.937 # SGD momentum
weight_decay: 0.0005 # optimizer weight decay
fl_gamma: 0.0 # focal loss gamma (efficientDet default is gamma=1.5)
hsv_h: 0.0138 # image HSV-Hue augmentation (fraction)
hsv_s: 0.678 # image HSV-Saturation augmentation (fraction)
hsv_v: 0.36 # image HSV-Value augmentation (fraction)
degrees: 0 # 1.98 * 0 # image rotation (+/- deg)
translate: 0 # 0.05 * 0 # image translation (+/- fraction)
scale: 0 #0 .05 * 0 # image scale (+/- gain)
shear: 0 # 0.641 * 0 # image shear (+/- deg)
experiments:
dir: ./experiments
detect:
2020-06-30 19:51:10 +00:00
source: /home/tomekb/yolov3/data/widok_01_19/widok_01_19_test_labels.txt
2020-07-02 16:10:02 +00:00
test-img-size: 1920
2020-06-28 12:37:21 +00:00
conf-thres: 0.3
iou-thres: 0.6
classes:
agnostic-nms:
augment:
confussion-matrix:
2020-06-30 19:51:10 +00:00
labels-dir: /home/tomekb/yolov3/data/widok_01_19/widok_01_19_labels
2020-06-28 12:37:21 +00:00
bayes:
todo: todo