Skip to content

Commit a309618

Browse files
Add MobileNetV5 to KerasHub (#2399)
* init: Add initial version * nit: I like the CI green, ig? * fix: Bring the numerics significantly closer * ✅ Yayy, 1e-5 is insanely good! * chore: Address Gemini reviews * review #1: Resolve reviews * feat: Add direct HF / TIMM checkpoint loading * refactor: Refactor handling of the MobileNetV5 architecture * nit: Updated examples in backbone and image classifier * fix: Save backbone and preprocessing, update docstring
1 parent 6f72208 commit a309618

18 files changed

+4212
-4
lines changed

keras_hub/api/layers/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,9 @@
108108
from keras_hub.src.models.mobilenet.mobilenet_image_converter import (
109109
MobileNetImageConverter as MobileNetImageConverter,
110110
)
111+
from keras_hub.src.models.mobilenetv5.mobilenetv5_image_converter import (
112+
MobileNetV5ImageConverter as MobileNetV5ImageConverter,
113+
)
111114
from keras_hub.src.models.moonshine.moonshine_audio_converter import (
112115
MoonshineAudioConverter as MoonshineAudioConverter,
113116
)

keras_hub/api/models/__init__.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -428,6 +428,15 @@
428428
from keras_hub.src.models.mobilenet.mobilenet_image_classifier_preprocessor import (
429429
MobileNetImageClassifierPreprocessor as MobileNetImageClassifierPreprocessor,
430430
)
431+
from keras_hub.src.models.mobilenetv5.mobilenetv5_backbone import (
432+
MobileNetV5Backbone as MobileNetV5Backbone,
433+
)
434+
from keras_hub.src.models.mobilenetv5.mobilenetv5_image_classifier import (
435+
MobileNetV5ImageClassifier as MobileNetV5ImageClassifier,
436+
)
437+
from keras_hub.src.models.mobilenetv5.mobilenetv5_image_classifier_preprocessor import (
438+
MobileNetV5ImageClassifierPreprocessor as MobileNetV5ImageClassifierPreprocessor,
439+
)
431440
from keras_hub.src.models.moonshine.moonshine_audio_to_text import (
432441
MoonshineAudioToText as MoonshineAudioToText,
433442
)

keras_hub/src/models/mobilenetv5/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)