update model name

mlexchange · Feb 20, 2024 · fecefeb · fecefeb
1 parent 74479a8
commit fecefeb
Showing 1 changed file with 26 additions and 26 deletions.
diff --git a/src/model_validation.py b/src/model_validation.py
@@ -4,36 +4,36 @@
 
 
 model_list_size = {
-    'xception' : 299,
-    'vgg16' : 224,
-    'vgg19' : 224,
-    'resnet101' : 224,
-    'resnet152' : 224,
-    'resnet50v2' : 224,
-    'resnet50': 224,
-    'resnet152v2' : 224,
-    'inceptionv3' : 299,
-    'densenet201' : 224,
-    'nasnetlarge' : 331,
-    'inceptionresnetv2' : 299,
-    'densenet169' : 224,
+    'Xception' : 299,
+    'VGG16' : 224,
+    'VGG19' : 224,
+    'ResNet101' : 224,
+    'ResNet152' : 224,
+    'ResNet50V2' : 224,
+    'ResNet50': 224,
+    'ResNet152V2' : 224,
+    'InceptionV3' : 299,
+    'DenseNet201' : 224,
+    'NASNetLarge' : 331,
+    'InceptionResNetV2' : 299,
+    'DenseNet169' : 224,
 }
 
 
 model_list_preprocess = {
-    'xception' : 'xception',
-    'vgg16' : 'vgg16',
-    'vgg19' : 'vgg19',
-    'resnet101' : 'resnet',
-    'resnet152' : 'resnet',
-    'resnet50v2' : 'resnet_v2',
-    'resnet50': 'resnet',
-    'resnet152v2' : 'resnet_v2',
-    'inceptionv3' : 'inception_v3',
-    'densenet201' : 'densenet',
-    'nasnetlarge' : 'nasnet',
-    'inceptionresnetv2' : 'inception_resnet_v2',
-    'densenet169' : 'densenet',
+    'Xception' : 'xception',
+    'VGG16' : 'vgg16',
+    'VGG19' : 'vgg19',
+    'ResNet101' : 'resnet',
+    'ResNet152' : 'resnet',
+    'ResNet50V2' : 'resnet_v2',
+    'ResNet50': 'resnet',
+    'ResNet152V2' : 'resnet_v2',
+    'InceptionV3' : 'inception_v3',
+    'DenseNet201' : 'densenet',
+    'NASNetLarge' : 'nasnet',
+    'InceptionResNetV2' : 'inception_resnet_v2',
+    'DenseNet169' : 'densenet',
 }