139 lines
2.3 KiB
YAML
139 lines
2.3 KiB
YAML
bayes:
|
|
iterations: 10
|
|
train:
|
|
epochs:
|
|
type: discrete
|
|
values: [30]
|
|
batch-size:
|
|
type: discrete
|
|
min: 1
|
|
max: 5
|
|
step: 1
|
|
cfg: ./cfg/yolov3-spp-21cls.cfg
|
|
data: ./data/widok_01_21.data
|
|
multi-scale:
|
|
type: discrete
|
|
values: [true, false]
|
|
img-size-start:
|
|
type: discrete
|
|
min: 512
|
|
max: 1088
|
|
step: 64
|
|
img-size-end:
|
|
type: discrete
|
|
min: 512
|
|
max: 1088
|
|
step: 64
|
|
rect:
|
|
type: discrete
|
|
values: [false]
|
|
resume: false
|
|
nosave: false
|
|
notest: false
|
|
evolve: false
|
|
bucket:
|
|
cache-images: false
|
|
weights: ./weights/yolov3-spp-ultralytics.pt
|
|
device: 1
|
|
adam:
|
|
type: discrete
|
|
values: [true]
|
|
single-cls: false
|
|
snapshot-every:
|
|
freeze-layers: true
|
|
other-hyps:
|
|
giou:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 10.0
|
|
cls:
|
|
type: continuous
|
|
min: 10.0
|
|
max: 100.0
|
|
cls_pw:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 10.0
|
|
obj:
|
|
type: continuous
|
|
min: 10.0
|
|
max: 100.0
|
|
obj_pw:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 10.0
|
|
iou_t:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 1.0
|
|
lr0:
|
|
type: continuous
|
|
min: 0.000001
|
|
max: 0.1
|
|
lrf:
|
|
type: continuous
|
|
min: 0.000001
|
|
max: 0.1
|
|
momentum:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 1.0
|
|
weight_decay:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 1.0
|
|
fl_gamma:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 10.0
|
|
hsv_h:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 1.0
|
|
hsv_s:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 1.0
|
|
hsv_v:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 1.0
|
|
degrees:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 30.0
|
|
translate:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 1.0
|
|
scale:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 1.0
|
|
shear:
|
|
type: continuous
|
|
min: 0.0
|
|
max: 1.0
|
|
experiments:
|
|
dir: ./experiments
|
|
detect:
|
|
source: ./data/widok_01_21/widok_01_21_test_labels.txt
|
|
test-img-size:
|
|
type: discrete
|
|
min: 512
|
|
max: 1088
|
|
step: 64
|
|
conf-thres:
|
|
type: continuous
|
|
min: 0.3
|
|
max: 0.6
|
|
iou-thres:
|
|
type: continuous
|
|
min: 0.3
|
|
max: 0.6
|
|
classes:
|
|
agnostic-nms:
|
|
augment:
|
|
confussion-matrix:
|
|
labels-dir: ./data/widok_01_21/widok_01_21_labels
|