Add MobileNet v2 and v3 artifact urls

* Add models to PRETRAINED_MODELS to enable acctest
* Update pretrained status in README

Author: paulnovo

Artifacts.toml

@@ -4,23 +4,23 @@ lazy = true
 
     [[mobilenet_v2.download]]
     sha256 = "24d8afb8897cb82825c51e65c7a02f4b2117581294de25ac7fd5222bbc15ae6e"
-    url = ""
+    url = "https://huggingface.co/FluxML/mobilenet/resolve/ccf1ffc5bbb4f2d3b9b7fb7e1285e52de337774e/mobilenet_v2-IMAGENET1K_V2.tar.gz"
 
 [mobilenet_v3_small]
 git-tree-sha1 = "9b1f0550051c731056cda7739fcdd115c36c04ad"
 lazy = true
 
     [[mobilenet_v3_small.download]]
     sha256 = "35b7f6d733bfbd1621349a0ec27391c2714add7ca80006fb4032d8bc66629c97"
-    url = ""
+    url = "https://huggingface.co/FluxML/mobilenet/resolve/ccf1ffc5bbb4f2d3b9b7fb7e1285e52de337774e/mobilenet_v3_small-IMAGENET1K_V1.tar.gz"
 
 [mobilenet_v3_large]
 git-tree-sha1 = "49971ff8327bc591885e78ff94140ee472f77329"
 lazy = true
 
     [[mobilenet_v3_large.download]]
     sha256 = "555fcb5f4f6574d77b603b2fc6672ab437ef60a1a20bc2f951122c91aaaf2f69"
-    url = ""
+    url = "https://huggingface.co/FluxML/mobilenet/resolve/ccf1ffc5bbb4f2d3b9b7fb7e1285e52de337774e/mobilenet_v3_large-IMAGENET1K_V2.tar.gz"
 
 [resnet101]
 git-tree-sha1 = "68d563526ab34d3e5aa66b7d96278d2acde212f9"

README.md

@@ -37,8 +37,8 @@ To contribute new models, see our [contributing docs](https://fluxml.ai/Metalhea
 | [InceptionResNet-v2](https://arxiv.org/abs/1602.07261) | [`InceptionResNetv2`](https://fluxml.ai/Metalhead.jl/dev/api/inception/#Metalhead.InceptionResNetv2) | N            |
 | [MLPMixer](https://arxiv.org/pdf/2105.01601)     | [`MLPMixer`](https://fluxml.ai/Metalhead.jl/dev/api/mixers/#Metalhead.MLPMixer)         | N            |
 | [MobileNetv1](https://arxiv.org/abs/1704.04861)  | [`MobileNetv1`](https://fluxml.ai/Metalhead.jl/dev/api/mobilenet/#Metalhead.MobileNetv1)   | N            |
-| [MobileNetv2](https://arxiv.org/abs/1801.04381)  | [`MobileNetv2`](https://fluxml.ai/Metalhead.jl/dev/api/mobilenet/#Metalhead.MobileNetv2)   | N            |
-| [MobileNetv3](https://arxiv.org/abs/1905.02244)  | [`MobileNetv3`](https://fluxml.ai/Metalhead.jl/dev/api/mobilenet/#Metalhead.MobileNetv3)   | N            |
+| [MobileNetv2](https://arxiv.org/abs/1801.04381)  | [`MobileNetv2`](https://fluxml.ai/Metalhead.jl/dev/api/mobilenet/#Metalhead.MobileNetv2)   | Y            |
+| [MobileNetv3](https://arxiv.org/abs/1905.02244)  | [`MobileNetv3`](https://fluxml.ai/Metalhead.jl/dev/api/mobilenet/#Metalhead.MobileNetv3)   | Y            |
 | [MNASNet](https://arxiv.org/abs/1807.11626)       | [`MNASNet`](https://fluxml.ai/Metalhead.jl/dev/api/efficientnet/#Metalhead.MNASNet)   | N            |
 | [ResMLP](https://arxiv.org/abs/2105.03404)       | [`ResMLP`](https://fluxml.ai/Metalhead.jl/dev/api/mixers/#Metalhead.ResMLP)                    | N            |
 | [ResNet](https://arxiv.org/abs/1512.03385)       | [`ResNet`](https://fluxml.ai/Metalhead.jl/dev/api/resnet/#Metalhead.ResNet)             | Y            |

src/convnets/mobilenets/mobilenetv2.jl

@@ -69,6 +69,9 @@ function MobileNetv2(width_mult::Real = 1; pretrain::Bool = false,
     layers = mobilenetv2(width_mult; inchannels, nclasses)
     model = MobileNetv2(layers)
     if pretrain
+        if width_mult != 1.0
+            throw(ArgumentError("No pre-trained weights available for width_mult=$width_mult."))
+        end
         loadpretrain!(model, "mobilenet_v2")
     end
     return model

src/convnets/mobilenets/mobilenetv3.jl

@@ -90,6 +90,9 @@ function MobileNetv3(config::Symbol; width_mult::Real = 1, pretrain::Bool = fals
     layers = mobilenetv3(config; width_mult, inchannels, nclasses)
     model = MobileNetv3(layers)
     if pretrain
+        if width_mult != 1.0
+            throw(ArgumentError("No pre-trained weights available for width_mult=$width_mult."))
+        end
         loadpretrain!(model, string("mobilenet_v3_", config))
     end
     return model

test/convnet_tests.jl

@@ -307,9 +307,9 @@ end
         m = MobileNetv2(width_mult) |> gpu
         @test size(m(x_224)) == (1000, 1)
         if (MobileNetv2, width_mult) in PRETRAINED_MODELS
-            @test acctest(MobileNetv2(; pretrain = true))
+            @test acctest(MobileNetv2(width_mult; pretrain = true))
         else
-            @test_throws ArgumentError MobileNetv2(pretrain = true)
+            @test_throws ArgumentError MobileNetv2(width_mult; pretrain = true)
         end
         @test gradtest(m, x_224)
     end
@@ -321,9 +321,9 @@ end
         m = MobileNetv3(config; width_mult) |> gpu
         @test size(m(x_224)) == (1000, 1)
         if (MobileNetv3, config, width_mult) in PRETRAINED_MODELS
-            @test acctest(MobileNetv3(config; pretrain = true))
+            @test acctest(MobileNetv3(config; width_mult, pretrain = true))
         else
-            @test_throws ArgumentError MobileNetv3(config; pretrain = true)
+            @test_throws ArgumentError MobileNetv3(config; width_mult, pretrain = true)
         end
         @test gradtest(m, x_224)
         _gc()

test/model_tests.jl

@@ -24,6 +24,9 @@ const PRETRAINED_MODELS = [
     # (DenseNet, 161),
     # (DenseNet, 169),
     # (DenseNet, 201),
+    (MobileNetv2, 1.0),
+    (MobileNetv3, :small, 1.0),
+    (MobileNetv3, :large, 1.0),
     (ResNet, 18),
     (ResNet, 34),
     (ResNet, 50),
@@ -85,4 +88,4 @@ end
 
 const x_224 = rand(Float32, 224, 224, 3, 1) |> gpu
 const x_256 = rand(Float32, 256, 256, 3, 1) |> gpu
-end
+end