tensorflow-image-models 0.0.8

Tensorflow image models

Install

pip install tensorflow_image_models==0.0.8

---

Usage

from tensorflow_image_models import list_models

list_models()

models list:
DPN92
DPN98
DPN131
DPN107
EfficientNetB0
EfficientNetB1
.....
.....
VGG16
VGG19

Create Model

from tensorflow_image_models import EfficientNet
model = EfficientNet.EfficientNetB0(classes=10)
model.summary()

from tensorflow_image_models import Inception
model = Inception.InceptionV4(classes=3, input_shape=(299,299,3)
model.summary()

---

Here is a list of input shape expected for each model

EfficientNetB0 224, 224, 3
EfficientNetB1 240, 240, 3
EfficientNetB2 260, 260, 3
EfficientNetB3 300, 300, 3
EfficientNetB4 380, 380, 3
EfficientNetB5 456, 456, 3
EfficientNetB6 528, 528, 3
EfficientNetB7 600, 600, 3
AlexNet 224, 224, 3
InceptionResNetV2 299, 299, 3
InceptionV3 299, 299, 3
LeNet 32, 32, 1
MobileNetV2 224, 224, 3
ResNet18 224, 224, 3
ResNet34 224, 224, 3
VoVNet27 224, 224, 3
VoVNet39 224, 224, 3
VoVNet57 224, 224, 3

The rest of the models receive any shape, and the default shape is (128,128,3)

---

License
This project is licensed under the MIT License

---

Author
DEEPOLOGY LAB