Skip to content

Commit adf1972

Browse files
authored
Merge pull request pytorch#726 from pytorch/v2-2021-5-26-torchvision-mobilenet-v3-implementation.md
Update 2021-5-26-torchvision-mobilenet-v3-implementation.md
2 parents 7a38d9c + 894e02d commit adf1972

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

_posts/2021-5-26-torchvision-mobilenet-v3-implementation.md

+4-2
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,8 @@ In this section, we will first provide benchmarks of the released models, and th
120120

121121
Here is how the models are initialized:
122122
```
123-
high_res = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_fpn(pretrained=T low_res = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_320_fpn(pretraine
123+
high_res = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_fpn(pretrained=True)
124+
low_res = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_320_fpn(pretrained=True)
124125
```
125126

126127
Below are some benchmarks between new and selected previous models. As we can see the high resolution Faster R-CNN with MobileNetV3-Large FPN backbone seems a viable replacement of the equivalent ResNet50 model for those users who are willing to sacrifice few accuracy points for a 5x speed-up:
@@ -166,7 +167,8 @@ In this section we will start by providing some benchmarks of the released pre-t
166167
This is how to initialize the pre-trained models:
167168

168169
```
169-
lraspp = torchvision.models.segmentation.lraspp_mobilenet_v3_large(pretrained=True) deeplabv3 = torchvision.models.segmentation.deeplabv3_mobilenet_v3_large(pretrained=Tr
170+
lraspp = torchvision.models.segmentation.lraspp_mobilenet_v3_large(pretrained=True)
171+
deeplabv3 = torchvision.models.segmentation.deeplabv3_mobilenet_v3_large(pretrained=True)
170172
```
171173

172174
Below are the detailed benchmarks between new and selected existing models. As we can see, the DeepLabV3 with a MobileNetV3-Large backbone is a viable replacement of FCN with ResNet50 for the majority of applications as it achieves similar accuracy with a 8.5x speed-up. We also observe that the LR-ASPP network supersedes the equivalent FCN in all metrics:

0 commit comments

Comments
 (0)