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