Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add shufflenetv2 1.5 and 2.0 weights #5906

Merged
merged 16 commits into from
Apr 28, 2022
Merged

Add shufflenetv2 1.5 and 2.0 weights #5906

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
d89f57a
Add shufflenetv2 1.5 and 2.0 weights
YosuaMichael Apr 27, 2022
c0516e7
Update recipe
YosuaMichael Apr 27, 2022
de2cf45
Add to docs
YosuaMichael Apr 27, 2022
5487b23
Use resize_size=232 for eval and update the result
YosuaMichael Apr 27, 2022
906fa30
Add quantized shufflenetv2 large
YosuaMichael Apr 27, 2022
de5d638
Update docs and readme
YosuaMichael Apr 27, 2022
7706d08
Format with ufmt
YosuaMichael Apr 27, 2022
06836b5
Add to hubconf.py
YosuaMichael Apr 27, 2022
b738c7d
Update readme for classification reference
YosuaMichael Apr 28, 2022
687498a
Merge branch 'main' into models/add-shufflenetv2-large
YosuaMichael Apr 28, 2022
de5a3b7
Fix reference classification readme
YosuaMichael Apr 28, 2022
3a2eec4
Merge branch 'main' into models/add-shufflenetv2-large
YosuaMichael Apr 28, 2022
dd97bc3
Fix typo on readme
YosuaMichael Apr 28, 2022
ef2dbc5
Merge branch 'main' into models/add-shufflenetv2-large
YosuaMichael Apr 28, 2022
c5551d7
Update reference/classification/readme
YosuaMichael Apr 28, 2022
691d8f6
Merge branch 'main' into models/add-shufflenetv2-large
YosuaMichael Apr 28, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion docs/source/models.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -176,8 +176,10 @@ Densenet-201 76.896 93.370
Densenet-161 77.138 93.560
Inception v3 77.294 93.450
GoogleNet 69.778 89.530
ShuffleNet V2 x1.0 69.362 88.316
ShuffleNet V2 x0.5 60.552 81.746
ShuffleNet V2 x1.0 69.362 88.316
ShuffleNet V2 x1.5 72.784 91.058
ShuffleNet V2 x2.0 76.200 92.888
MobileNet V2 71.878 90.286
MobileNet V3 Large 74.042 91.340
MobileNet V3 Small 67.668 87.402
Expand Down
30 changes: 28 additions & 2 deletions torchvision/models/shufflenetv2.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -223,11 +223,37 @@ class ShuffleNet_V2_X1_0_Weights(WeightsEnum):


class ShuffleNet_V2_X1_5_Weights(WeightsEnum):
pass
IMAGENET1K_V1 = Weights(
url="https://download.pytorch.org/models/shufflenetv2_x1_5-3c479a10.pth",
transforms=partial(ImageClassification, crop_size=224),
YosuaMichael marked this conversation as resolved.
Show resolved Hide resolved
meta={
**_COMMON_META,
"recipe": "https://github.com/pytorch/vision/pull/5906",
"num_params": 3503624,
"metrics": {
"acc@1": 72.784,
"acc@5": 91.058,
},
},
)
DEFAULT = IMAGENET1K_V1


class ShuffleNet_V2_X2_0_Weights(WeightsEnum):
pass
IMAGENET1K_V1 = Weights(
url="https://download.pytorch.org/models/shufflenetv2_x2_0-8be3c8ee.pth",
transforms=partial(ImageClassification, crop_size=224),
meta={
**_COMMON_META,
"recipe": "https://github.com/pytorch/vision/pull/5906",
"num_params": 7393996,
"metrics": {
"acc@1": 76.200,
"acc@5": 92.888,
},
},
)
DEFAULT = IMAGENET1K_V1


@handle_legacy_interface(weights=("pretrained", ShuffleNet_V2_X0_5_Weights.IMAGENET1K_V1))
Expand Down