car-detection-bayes/our_scripts/config_bayes.yml

105 lines
1.9 KiB
YAML
Raw Normal View History

2020-07-25 22:43:07 +00:00
bayes:
2020-09-11 16:52:00 +00:00
iterations: 1
2020-07-25 22:42:33 +00:00
train:
2020-07-25 22:43:07 +00:00
epochs:
type: discrete
2020-09-11 16:52:00 +00:00
values: [400]
2020-07-25 22:42:33 +00:00
batch-size:
type: discrete
2020-09-11 16:52:00 +00:00
values: [4]
2020-07-25 22:43:07 +00:00
cfg: ./cfg/yolov3-spp-21cls.cfg
data: ./data/widok_01_21.data
2020-07-25 22:42:33 +00:00
multi-scale:
type: discrete
2020-09-11 16:52:00 +00:00
values: [false]
2020-07-25 22:43:07 +00:00
img-size-start:
type: discrete
2020-09-11 16:52:00 +00:00
values: [896]
2020-07-25 22:43:07 +00:00
img-size-end:
2020-07-25 22:42:33 +00:00
type: discrete
2020-09-11 16:52:00 +00:00
values: [1344]
2020-07-25 22:43:07 +00:00
rect:
2020-07-25 22:42:33 +00:00
type: discrete
2020-07-25 22:43:07 +00:00
values: [false]
2020-07-25 22:42:33 +00:00
resume: false
nosave: false
notest: false
evolve: false
bucket:
cache-images: false
2020-09-11 16:52:00 +00:00
weights: ./experiments/model2/best.pt
2020-07-25 22:42:33 +00:00
device: 1
2020-07-25 22:43:07 +00:00
adam:
type: discrete
values: [true]
2020-07-25 22:42:33 +00:00
single-cls: false
2020-07-25 22:43:07 +00:00
snapshot-every:
2020-09-11 16:52:00 +00:00
freeze-layers: false
2020-07-25 22:42:33 +00:00
other-hyps:
giou:
2020-09-11 16:52:00 +00:00
type: discrete
values: [3.54]
2020-07-25 22:42:33 +00:00
cls:
2020-09-11 16:52:00 +00:00
type: discrete
values: [3.74]
2020-07-25 22:42:33 +00:00
cls_pw:
2020-09-11 16:52:00 +00:00
type: discrete
values: [1.0]
2020-07-25 22:42:33 +00:00
obj:
2020-09-11 16:52:00 +00:00
type: discrete
values: [64.3]
2020-07-25 22:42:33 +00:00
obj_pw:
2020-09-11 16:52:00 +00:00
type: discrete
values: [1.0]
2020-07-25 22:42:33 +00:00
iou_t:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.2]
2020-07-25 22:43:07 +00:00
lr0:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.01]
2020-07-25 22:43:07 +00:00
lrf:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.0005]
2020-07-25 22:42:33 +00:00
momentum:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.937]
2020-07-25 22:42:33 +00:00
weight_decay:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.0005]
2020-07-25 22:42:33 +00:00
fl_gamma:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.0]
2020-07-25 22:42:33 +00:00
hsv_h:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.0138]
2020-07-25 22:43:07 +00:00
hsv_s:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.678]
2020-07-25 22:42:33 +00:00
hsv_v:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.36]
2020-07-25 22:42:33 +00:00
degrees:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.0]
2020-07-25 22:42:33 +00:00
translate:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.0]
2020-07-25 22:42:33 +00:00
scale:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.0]
2020-07-25 22:42:33 +00:00
shear:
2020-09-11 16:52:00 +00:00
type: discrete
values: [0.0]
2020-07-25 22:42:33 +00:00
experiments:
dir: ./experiments
detect:
2020-07-25 22:43:07 +00:00
source: ./data/widok_01_21/widok_01_21_test_labels.txt
2020-09-11 16:52:00 +00:00
test-img-size: 1024
conf-thres: 0.45
iou-thres: 0.6
2020-07-25 22:42:33 +00:00
classes:
agnostic-nms:
augment:
confussion-matrix:
2020-07-25 22:43:07 +00:00
labels-dir: ./data/widok_01_21/widok_01_21_labels