Skip to content

GitLab

"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "591c899c6e9272e5a3c0cd2e3bddf82d5fad645e"
Unverified Commit 7e2050f1 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix hashes of efficientnet weights (#8036)

parent 0f749070
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
torchvision/models/efficientnet.py
View file @ 7e2050f1
...@@ -439,7 +439,7 @@ _COMMON_META_V2 = { ...@@ -439,7 +439,7 @@ _COMMON_META_V2 = {
class EfficientNet_B0_Weights(WeightsEnum): class EfficientNet_B0_Weights(WeightsEnum):
IMAGENET1K_V1 = Weights( IMAGENET1K_V1 = Weights(
# Weights ported from https://github.com/rwightman/pytorch-image-models/ # Weights ported from https://github.com/rwightman/pytorch-image-models/
url="https://download.pytorch.org/models/efficientnet_b0_rwightman-3dd342df.pth", url="https://download.pytorch.org/models/efficientnet_b0_rwightman-7f5810bc.pth",
transforms=partial( transforms=partial(
ImageClassification, crop_size=224, resize_size=256, interpolation=InterpolationMode.BICUBIC ImageClassification, crop_size=224, resize_size=256, interpolation=InterpolationMode.BICUBIC
), ),
...@@ -511,7 +511,7 @@ class EfficientNet_B1_Weights(WeightsEnum): ...@@ -511,7 +511,7 @@ class EfficientNet_B1_Weights(WeightsEnum):
class EfficientNet_B2_Weights(WeightsEnum): class EfficientNet_B2_Weights(WeightsEnum):
IMAGENET1K_V1 = Weights( IMAGENET1K_V1 = Weights(
# Weights ported from https://github.com/rwightman/pytorch-image-models/ # Weights ported from https://github.com/rwightman/pytorch-image-models/
url="https://download.pytorch.org/models/efficientnet_b2_rwightman-bcdf34b7.pth", url="https://download.pytorch.org/models/efficientnet_b2_rwightman-c35c1473.pth",
transforms=partial( transforms=partial(
ImageClassification, crop_size=288, resize_size=288, interpolation=InterpolationMode.BICUBIC ImageClassification, crop_size=288, resize_size=288, interpolation=InterpolationMode.BICUBIC
), ),
...@@ -535,7 +535,7 @@ class EfficientNet_B2_Weights(WeightsEnum): ...@@ -535,7 +535,7 @@ class EfficientNet_B2_Weights(WeightsEnum):
class EfficientNet_B3_Weights(WeightsEnum): class EfficientNet_B3_Weights(WeightsEnum):
IMAGENET1K_V1 = Weights( IMAGENET1K_V1 = Weights(
# Weights ported from https://github.com/rwightman/pytorch-image-models/ # Weights ported from https://github.com/rwightman/pytorch-image-models/
url="https://download.pytorch.org/models/efficientnet_b3_rwightman-cf984f9c.pth", url="https://download.pytorch.org/models/efficientnet_b3_rwightman-b3899882.pth",
transforms=partial( transforms=partial(
ImageClassification, crop_size=300, resize_size=320, interpolation=InterpolationMode.BICUBIC ImageClassification, crop_size=300, resize_size=320, interpolation=InterpolationMode.BICUBIC
), ),
...@@ -559,7 +559,7 @@ class EfficientNet_B3_Weights(WeightsEnum): ...@@ -559,7 +559,7 @@ class EfficientNet_B3_Weights(WeightsEnum):
class EfficientNet_B4_Weights(WeightsEnum): class EfficientNet_B4_Weights(WeightsEnum):
IMAGENET1K_V1 = Weights( IMAGENET1K_V1 = Weights(
# Weights ported from https://github.com/rwightman/pytorch-image-models/ # Weights ported from https://github.com/rwightman/pytorch-image-models/
url="https://download.pytorch.org/models/efficientnet_b4_rwightman-7eb33cd5.pth", url="https://download.pytorch.org/models/efficientnet_b4_rwightman-23ab8bcd.pth",
transforms=partial( transforms=partial(
ImageClassification, crop_size=380, resize_size=384, interpolation=InterpolationMode.BICUBIC ImageClassification, crop_size=380, resize_size=384, interpolation=InterpolationMode.BICUBIC
), ),
...@@ -583,7 +583,7 @@ class EfficientNet_B4_Weights(WeightsEnum): ...@@ -583,7 +583,7 @@ class EfficientNet_B4_Weights(WeightsEnum):
class EfficientNet_B5_Weights(WeightsEnum): class EfficientNet_B5_Weights(WeightsEnum):
IMAGENET1K_V1 = Weights( IMAGENET1K_V1 = Weights(
# Weights ported from https://github.com/lukemelas/EfficientNet-PyTorch/ # Weights ported from https://github.com/lukemelas/EfficientNet-PyTorch/
url="https://download.pytorch.org/models/efficientnet_b5_lukemelas-b6417697.pth", url="https://download.pytorch.org/models/efficientnet_b5_lukemelas-1a07897c.pth",
transforms=partial( transforms=partial(
ImageClassification, crop_size=456, resize_size=456, interpolation=InterpolationMode.BICUBIC ImageClassification, crop_size=456, resize_size=456, interpolation=InterpolationMode.BICUBIC
), ),
...@@ -607,7 +607,7 @@ class EfficientNet_B5_Weights(WeightsEnum): ...@@ -607,7 +607,7 @@ class EfficientNet_B5_Weights(WeightsEnum):
class EfficientNet_B6_Weights(WeightsEnum): class EfficientNet_B6_Weights(WeightsEnum):
IMAGENET1K_V1 = Weights( IMAGENET1K_V1 = Weights(
# Weights ported from https://github.com/lukemelas/EfficientNet-PyTorch/ # Weights ported from https://github.com/lukemelas/EfficientNet-PyTorch/
url="https://download.pytorch.org/models/efficientnet_b6_lukemelas-c76e70fd.pth", url="https://download.pytorch.org/models/efficientnet_b6_lukemelas-24a108a5.pth",
transforms=partial( transforms=partial(
ImageClassification, crop_size=528, resize_size=528, interpolation=InterpolationMode.BICUBIC ImageClassification, crop_size=528, resize_size=528, interpolation=InterpolationMode.BICUBIC
), ),
...@@ -631,7 +631,7 @@ class EfficientNet_B6_Weights(WeightsEnum): ...@@ -631,7 +631,7 @@ class EfficientNet_B6_Weights(WeightsEnum):
class EfficientNet_B7_Weights(WeightsEnum): class EfficientNet_B7_Weights(WeightsEnum):
IMAGENET1K_V1 = Weights( IMAGENET1K_V1 = Weights(
# Weights ported from https://github.com/lukemelas/EfficientNet-PyTorch/ # Weights ported from https://github.com/lukemelas/EfficientNet-PyTorch/
url="https://download.pytorch.org/models/efficientnet_b7_lukemelas-dcc49843.pth", url="https://download.pytorch.org/models/efficientnet_b7_lukemelas-c5b4e57e.pth",
transforms=partial( transforms=partial(
ImageClassification, crop_size=600, resize_size=600, interpolation=InterpolationMode.BICUBIC ImageClassification, crop_size=600, resize_size=600, interpolation=InterpolationMode.BICUBIC
), ),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment