-
Notifications
You must be signed in to change notification settings - Fork 23
/
params.yaml
32 lines (28 loc) · 962 Bytes
/
params.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
base:
project: defect_detection
pkg_list_fname: pkg_list.txt
random_state: 0
data_load:
dataset_url: https://github.com/abin24/Magnetic-tile-defect-datasets./archive/refs/heads/master.zip
data_dir: data
orig_dirname: Magnetic-tile-defect-datasets.-master
new_dirname: MAGNETIC_TILE_SURFACE_DEFECTS
data_split:
test_pct: 0.20
train_img_dir_path: data/MAGNETIC_TILE_SURFACE_DEFECTS/train_images
train_mask_dir_path: data/MAGNETIC_TILE_SURFACE_DEFECTS/train_masks
test_img_dir_path: data/MAGNETIC_TILE_SURFACE_DEFECTS/test_images
test_mask_dir_path: data/MAGNETIC_TILE_SURFACE_DEFECTS/test_masks
train:
img_size: 224
valid_pct: 0.2
learning_rate: 0.02
batch_size: 16
n_epochs: 6
use_cpu: false
model_pickle_fpath: models/model_pickle_fastai.pkl
augmentations: {do_flip: false, flip_vert: false, max_rotate: 10.0, min_zoom: 1.0,
max_zoom: 1.0}
evaluate:
save_test_preds: true
metrics_file: evaluation/metrics.json