lucasoskorep/tensordex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: adding resnet and formatting updates

Browse Source
This commit is contained in:
Lucas Oskorep
2023-04-06 00:37:46 -04:00
parent ce5939d8a9
commit dc427837f6
12 changed files with 87 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modeling_utils/image_class_builder.py
+10 -1
View File
@@ -15,11 +15,21 @@ class ImageClassModels(Enum):
keras.applications.inception_v3.preprocess_input,
"inception_v3"
)
INCEPTION_RESNET_V2 = ModelWrapper(
keras.applications.inception_resnet_v2.InceptionResNetV2,
keras.applications.inception_resnet_v2.preprocess_input,
"inception_resnet_v2"
)
XCEPTION = ModelWrapper(
keras.applications.xception.Xception,
keras.applications.xception.preprocess_input,
"xception"
)
DENSENET201 = ModelWrapper(
keras.applications.densenet.DenseNet201,
keras.applications.densenet.preprocess_input,
"densenet201"
)
MOBILENET_V2 = ModelWrapper(
keras.applications.mobilenet_v2.MobileNetV2,
keras.applications.mobilenet_v2.preprocess_input,
@@ -34,7 +44,6 @@ class ImageClassModels(Enum):
keras.applications.efficientnet_v2.EfficientNetV2B0,
tf.keras.applications.efficientnet_v2.preprocess_input,
"efficientnet_v2b0"
)