Skip to content

Commit 8b5e75b

Browse files
committed
update config, remove default params
1 parent 4d3d723 commit 8b5e75b

File tree

2 files changed

+4
-32
lines changed

2 files changed

+4
-32
lines changed

classification/AntBee/config/train_test_ce1.cfg

Lines changed: 2 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -18,30 +18,17 @@ train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd]
1818
valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
1919
test_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
2020

21-
Rescale_output_size = [256, 256]
22-
Rescale_inverse = False
23-
21+
Rescale_output_size = [256, 256]
2422
RandomCrop_output_size = [224, 224]
25-
RandomCrop_foreground_focus = False
26-
RandomCrop_foreground_ratio = None
27-
RandomCrop_mask_label = None
28-
RandomCrop_inverse = False
29-
3023
CenterCrop_output_size = [224, 224]
31-
CenterCrop_inverse = False
3224

3325
RandomFlip_flip_depth = False
3426
RandomFlip_flip_height = False
3527
RandomFlip_flip_width = True
36-
RandomFlip_inverse = False
3728

29+
NormalizeWithMeanStd_channels = [0, 1, 2]
3830
NormalizeWithMeanStd_mean = [123.675, 116.28, 103.53]
3931
NormalizeWithMeanStd_std = [58.395, 57.12, 57.375]
40-
NormalizeWithMeanStd_channels = [0, 1, 2]
41-
NormalizeWithMeanStd_mask = False
42-
NormalizeWithMeanStd_random_fill = False
43-
NormalizeWithMeanStd_inverse = False
44-
4532

4633
[network]
4734
# this section gives parameters for network
@@ -89,5 +76,4 @@ gpus = [0]
8976
# checkpoint mode can be [0-latest, 1-best, 2-specified]
9077
ckpt_mode = 1
9178
output_csv = result/resnet18_ce1.csv
92-
evaluation_mode = True
9379
save_probability = True

classification/AntBee/config/train_test_ce2.cfg

Lines changed: 2 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -18,30 +18,17 @@ train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd]
1818
valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
1919
test_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
2020

21-
Rescale_output_size = [256, 256]
22-
Rescale_inverse = False
23-
21+
Rescale_output_size = [256, 256]
2422
RandomCrop_output_size = [224, 224]
25-
RandomCrop_foreground_focus = False
26-
RandomCrop_foreground_ratio = None
27-
RandomCrop_mask_label = None
28-
RandomCrop_inverse = False
29-
3023
CenterCrop_output_size = [224, 224]
31-
CenterCrop_inverse = False
3224

3325
RandomFlip_flip_depth = False
3426
RandomFlip_flip_height = False
3527
RandomFlip_flip_width = True
36-
RandomFlip_inverse = False
3728

29+
NormalizeWithMeanStd_channels = [0, 1, 2]
3830
NormalizeWithMeanStd_mean = [123.675, 116.28, 103.53]
3931
NormalizeWithMeanStd_std = [58.395, 57.12, 57.375]
40-
NormalizeWithMeanStd_channels = [0, 1, 2]
41-
NormalizeWithMeanStd_mask = False
42-
NormalizeWithMeanStd_random_fill = False
43-
NormalizeWithMeanStd_inverse = False
44-
4532

4633
[network]
4734
# this section gives parameters for network
@@ -90,5 +77,4 @@ gpus = [0]
9077
# checkpoint mode can be [0-latest, 1-best, 2-specified]
9178
ckpt_mode = 1
9279
output_csv = result/resnet18_ce2.csv
93-
evaluation_mode = True
9480
save_probability = True

0 commit comments

Comments
 (0)