cameronraysmith
/
magnetic-tiles-defect
connected to https://github.com/cameronraysmith/magnetic-tiles-defect.git


  
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: 30
  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