Skip to content

Commit 459b48e

Browse files
committed
Add MobileNetV3small as a backbone
1 parent c1cfb14 commit 459b48e

2 files changed

Lines changed: 175 additions & 90 deletions

File tree

‎README.md‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -63,11 +63,11 @@ Please check that **Tensorflow** is installed on your computer.
6363

6464
To import the library just use the standard python import statement:
6565

66-
66+
6767
```python
6868
import tensorflow_advanced_segmentation_models as tasm
6969
```
70-
70+
7171
Then pick any model backbone from the list below and define weights, height and width:
7272

7373
```python
@@ -90,7 +90,7 @@ base_model, layers, layer_names = tasm.create_base_model(name=BACKBONE_NAME, wei
9090
```
9191

9292
Define a Model and compile it with an appropriate loss:
93-
93+
9494
```python
9595
model = tasm.DANet(n_classes=3, base_model=base_model, output_layers=layers, backbone_trainable=False)
9696
model.compile(tf.keras.optimizers.Adam(0.0001), loss=tasm.losses.CategoricalFocalLoss, tasm.metrics.IOUScore(threshold=0.5))
@@ -113,7 +113,7 @@ history = model.fit(
113113
validation_data=ValidationGenerator
114114
)
115115
```
116-
116+
117117
You can use the fit_generator method too, e.g. if you want to apply augmentations to the data.
118118
For complete training pipelines, go to the <a href="https://github.com/JanMarcelKezmann/TensorFlow-Advanced-Segmentation-Models/blob/master/examples">Examples</a> folder
119119

@@ -142,7 +142,7 @@ For complete training pipelines, go to the <a href="https://github.com/JanMarcel
142142
- **<a href="https://arxiv.org/pdf/1909.11065.pdf">ASPOCRNet</a>** &nbsp; <a href="https://github.com/JanMarcelKezmann/TensorFlow-Advanced-Segmentation-Models/blob/master/tensorflow_advanced_segmentation_models/models/ASPOCRNet.py"><img align="center" width="20px" src="https://cdn.iconscout.com/icon/free/png-512/code-280-460136.png" /></a>
143143
- <a href="https://arxiv.org/pdf/1909.09408.pdf">**ACFNet**</a> &nbsp; <a href="https://github.com/JanMarcelKezmann/TensorFlow-Advanced-Segmentation-Models/blob/master/tensorflow_advanced_segmentation_models/models/ACFNet.py"><img align="center" width="20px" src="https://cdn.iconscout.com/icon/free/png-512/code-280-460136.png" /></a>
144144
- <a href="https://arxiv.org/pdf/1904.04514.pdf">**HRNet + OCR**</a> &nbsp; <a href="https://github.com/JanMarcelKezmann/TensorFlow-Advanced-Segmentation-Models/blob/master/tensorflow_advanced_segmentation_models/models/HRNetOCR.py"><img align="center" width="20px" src="https://cdn.iconscout.com/icon/free/png-512/code-280-460136.png" /></a> **NEW!**
145-
145+
146146
**Backbones**
147147
(For Details see <a href="https://github.com/JanMarcelKezmann/TensorFlow-Advanced-Segmentation-Models/tree/master/tensorflow_advanced_segmentation_models/backbones">here</a>.)
148148

@@ -151,14 +151,14 @@ For complete training pipelines, go to the <a href="https://github.com/JanMarcel
151151
|**VGG** | ``'vgg16' 'vgg19'``|
152152
|**ResNet** | ``'resnet50' 'resnet50v2' 'resnet101' 'resnet101v2' 'resnet152' 'resnet152v2'``|
153153
|**Xception** | ``'xception'``|
154-
|**MobileNet** | ``'mobilenet' 'mobilenetv2'``|
154+
|**MobileNet** | ``'mobilenet' 'mobilenetv2' 'mobilenetv3small'``|
155155
|**NASNet** | ``'nasnetlarge' 'nasnetmobile'``|
156156
|**DenseNet** | ``'densenet121' 'densenet169' 'densenet201'``|
157157
|**EfficientNet** | ``'efficientnetb0' 'efficientnetb1' 'efficientnetb2' 'efficientnetb3' 'efficientnetb4' 'efficientnetb5' 'efficientnetb6' efficientnetb7'``|
158-
158+
159159

160160
All backbones have weights trained on 2012 ILSVRC ImageNet dataset.
161-
161+
162162
**Further Model Information**
163163

164164
A new feature makes it possible to define the model as a Subclassed Model or as a Functional Model instead. To define the model as a Subclassed Model just write: **tasm.UNet** to define the UNet or replace it with any other model. If you want to define the Functional Model instead just append **.model()**, i.e. **tasm.UNet.model()**. This provides further TensorFlow features like saving the model in the "tf" format.
@@ -172,8 +172,8 @@ A new feature makes it possible to define the model as a Subclassed Model or as
172172
Publisher = {GitHub},
173173
Journal = {GitHub repository},
174174
Howpublished = {\url{https://github.com/JanMarcelKezmann/TensorFlow-Advanced-Segmentation-Models}}
175-
}
176-
175+
}
176+
177177
## License
178178

179179
Project is distributed under <a href="https://github.com/JanMarcelKezmann/TensorFlow-Advanced-Segmentation-Models/blob/master/LICENSE">MIT License</a>.

0 commit comments

Comments
 (0)