Image Classification
annanau commited on
Commit
3ca67a9
·
verified ·
1 Parent(s): e728598

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. inceptionV3.yaml +23 -0
  2. resnet50.yaml +24 -0
  3. xception.yaml +23 -0
inceptionV3.yaml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ DATASET:
2
+ data_path: "./data/experiment_5/prop_0_0"
3
+ train_path: "./data/experiment_5/prop_0_0/train_80"
4
+ val_path: "./data/experiment_5/prop_0_0/val_10"
5
+ test_path: "./data/experiment_5/prop_0_0/test_10"
6
+ num_class: 4 # 0:1-4, 1:5-7, 2:8-9, 3:10-13
7
+ img_size: 299
8
+
9
+ MODEL:
10
+ name: "inceptionV3"
11
+ ckpt_path: "./ckpt/experiment_5/"
12
+ model_path: "./models/experiment_5/inceptionV3_prop_0_0"
13
+ plots_path: "./plots/experiment_5/"
14
+
15
+ TRAIN:
16
+ batch_size: 32
17
+ num_epoch: 500
18
+ augment: True
19
+ tune: False
20
+ optimizer: "Adam"
21
+ lr: 0.001
22
+ lr_tune: 0.0001
23
+ es: 15
resnet50.yaml ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ DATASET:
2
+ data_path: "./data/head/experiment_5/prop_0_0/"
3
+ train_path: "./data/head/experiment_5/prop_0_0/train_80"
4
+ val_path: "./data/head/experiment_5/prop_0_0/val_10"
5
+ test_path: "./data/head/experiment_5/prop_0_0/test_10"
6
+ num_class: 4
7
+ img_size: 299
8
+
9
+ MODEL:
10
+ name: "resnet50"
11
+ ckpt_path: "./ckpt/head/experiment_5/"
12
+ model_path: "./models/head/experiment_5/"
13
+ preds_path: "./predictions/head/experiment_5/"
14
+ plots_path: "./plots/head/experiment_5/"
15
+
16
+ TRAIN:
17
+ batch_size: 32
18
+ num_epoch: 500
19
+ augment: True
20
+ tune: True
21
+ optimizer: "Adam"
22
+ lr: 0.001
23
+ lr_tune: 0.0001
24
+ es: 20
xception.yaml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ DATASET:
2
+ data_path: "./data/head/experiment_5/prop_0_0/"
3
+ train_path: "./data/head/experiment_5/prop_0_0/train_80"
4
+ val_path: "./data/head/experiment_5/prop_0_0/val_10"
5
+ test_path: "./data/head/experiment_5/prop_0_0/test_10"
6
+ num_class: 4
7
+ img_size: 299
8
+
9
+ MODEL:
10
+ name: "xception"
11
+ ckpt_path: "./ckpt/head/experiment_5/"
12
+ model_path: "./models/head/experiment_5/"
13
+ preds_path: "./predictions/head/experiment_5/"
14
+ plots_path: "./plots/head/experiment_5/"
15
+
16
+ TRAIN:
17
+ batch_size: 32
18
+ num_epoch: 200
19
+ augment: True
20
+ optimizer: "Adam"
21
+ lr: 0.001
22
+ lr_tune: 0.0001
23
+ es: 20