update config for classification

yeyeyeping · yeyeyeping · commit 49b29395c379 · 2023-02-13T22:05:06.000+08:00
diff --git a/classification/AntBee/config/train_test_ce1.cfg b/classification/AntBee/config/train_test_ce1.cfg
@@ -14,8 +14,8 @@ train_batch_size = 4
 modal_num = 1
 
 # data transforms
-train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd]
-valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
+train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd, LabelToProbability]
+valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd, LabelToProbability]
 test_transform  = [Rescale, CenterCrop, NormalizeWithMeanStd]
 
 Rescale_output_size    = [256, 256]
diff --git a/classification/AntBee/config/train_test_ce2.cfg b/classification/AntBee/config/train_test_ce2.cfg
@@ -14,8 +14,8 @@ train_batch_size = 4
 modal_num = 1
 
 # data transforms
-train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd]
-valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
+train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd, LabelToProbability]
+valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd, LabelToProbability]
 test_transform  = [Rescale, CenterCrop, NormalizeWithMeanStd]
 
 Rescale_output_size    = [256, 256]
diff --git a/classification/CHNCXR/config/net_resnet18.cfg b/classification/CHNCXR/config/net_resnet18.cfg
@@ -14,8 +14,8 @@ train_batch_size = 4
 modal_num = 1
 
 # data transforms
-train_transform = [Rescale, RandomCrop, RandomFlip, GrayscaleToRGB, NormalizeWithMeanStd]
-valid_transform = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd]
+train_transform = [Rescale, RandomCrop, RandomFlip, GrayscaleToRGB, NormalizeWithMeanStd, LabelToProbability]
+valid_transform = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd, LabelToProbability]
 test_transform  = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd]
 
 Rescale_output_size    = [256, 256]
diff --git a/classification/CHNCXR/config/net_vgg16.cfg b/classification/CHNCXR/config/net_vgg16.cfg
@@ -14,8 +14,8 @@ train_batch_size = 4
 modal_num = 1
 
 # data transforms
-train_transform = [Rescale, RandomCrop, RandomFlip, GrayscaleToRGB, NormalizeWithMeanStd]
-valid_transform = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd]
+train_transform = [Rescale, RandomCrop, RandomFlip, GrayscaleToRGB, NormalizeWithMeanStd, LabelToProbability]
+valid_transform = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd, LabelToProbability]
 test_transform  = [Rescale, CenterCrop, GrayscaleToRGB, NormalizeWithMeanStd]
 
 Rescale_output_size    = [256, 256]
