Skip to content

Commit 75be51b

Browse files
authored
Update 2021-4-15-ML-models-TorchVision-v0.9.md
1 parent 965d1da commit 75be51b

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

_posts/2021-4-15-ML-models-TorchVision-v0.9.md

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@ TorchVision v0.9 has been [released](https://github.com/pytorch/vision/releases)
1111
* **Quantized MobileNetV3 Large:** The quantized version of MobilNetV3 Large reduces the number of parameters by 45% and it is roughly 2.5x faster than the non-quantized version while remaining competitive in [terms of accuracy](https://github.com/pytorch/vision/blob/master/docs/source/models.rst#quantized-models). It was fitted on ImageNet using Quantization Aware Training by iterating on the non-quantized version and it can be trained from scratch using the existing [reference scripts](https://github.com/pytorch/vision/tree/master/references/classification#quantized).
1212

1313
**Usage:**
14-
```model = torchvision.models.mobilenet_v3_large(pretrained=True)
14+
```
15+
model = torchvision.models.mobilenet_v3_large(pretrained=True)
1516
# model = torchvision.models.mobilenet_v3_small(pretrained=True)
1617
# model = torchvision.models.quantization.mobilenet_v3_large(pretrained=True)
1718
model.eval()
@@ -22,7 +23,8 @@ predictions = model(img)
2223
* **Faster R-CNN MobileNetV3-Large 320 FPN:** This is an iteration of the previous model that uses reduced resolution (min_size=320 pixel) and sacrifices accuracy for speed. It is 25x faster on CPU than the equivalent ResNet50 detector and thus it is good for real mobile use-cases.
2324

2425
**Usage:**
25-
```model = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_fpn(pretrained=True)
26+
```
27+
model = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_fpn(pretrained=True)
2628
# model = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_320_fpn(pretrained=True)
2729
model.eval()
2830
predictions = model(img)
@@ -32,7 +34,8 @@ predictions = model(img)
3234
* **Lite R-ASPP with Dilated MobileNetV3 Large Backbone:** We introduce the implementation of a new segmentation head called Lite R-ASPP and combine it with the dilated MobileNetV3 Large backbone to build a very fast segmentation model. The new model sacrifices some accuracy to achieve a 15x speed improvement comparing to the previously most lightweight segmentation model which was the FCN ResNet50.
3335

3436
**Usage:**
35-
```model = torchvision.models.segmentation.deeplabv3_mobilenet_v3_large(pretrained=True)
37+
```
38+
model = torchvision.models.segmentation.deeplabv3_mobilenet_v3_large(pretrained=True)
3639
# model = torchvision.models.segmentation.lraspp_mobilenet_v3_large(pretrained=True)
3740
model.eval()
3841
predictions = model(img)

0 commit comments

Comments
 (0)