Add Inception_ResNetv2

Author: theabhirath

src/Metalhead.jl

@@ -1,7 +1,7 @@
 module Metalhead
 
 using Flux
-using Flux: outputsize, Zygote
+using Flux: Zygote, outputsize
 using Functors
 using BSON
 using Artifacts, LazyArtifacts
@@ -19,8 +19,7 @@ using .Layers
 # CNN models
 include("convnets/alexnet.jl")
 include("convnets/vgg.jl")
-include("convnets/inceptionv3.jl")
-include("convnets/inceptionv4.jl")
+include("convnets/inception.jl")
 include("convnets/googlenet.jl")
 include("convnets/resnet.jl")
 include("convnets/resnext.jl")
@@ -40,20 +39,20 @@ include("pretrain.jl")
 
 export AlexNet,
        VGG, VGG11, VGG13, VGG16, VGG19,
+       GoogLeNet,
        ResNet, ResNet18, ResNet34, ResNet50, ResNet101, ResNet152,
-       GoogLeNet, Inception3, SqueezeNet,
-       DenseNet, DenseNet121, DenseNet161, DenseNet169, DenseNet201,
+       Inception3, Inception4, SqueezeNet,
        ResNeXt,
+       DenseNet, DenseNet121, DenseNet161, DenseNet169, DenseNet201,
        MobileNetv1, MobileNetv2, MobileNetv3,
        MLPMixer, ResMLP, gMLP,
        ViT,
-       ConvNeXt, ConvMixer
+       ConvMixer, ConvNeXt
 
 # use Flux._big_show to pretty print large models
-for T in (:AlexNet, :VGG, :ResNet, :GoogLeNet, :Inception3, :SqueezeNet, :DenseNet,
-          :ResNeXt,
-          :MobileNetv1, :MobileNetv2, :MobileNetv3,
-          :MLPMixer, :ResMLP, :gMLP, :ViT, :ConvNeXt, :ConvMixer)
+for T in (:AlexNet, :VGG, :GoogLeNet, :ResNet, :ResNeXt, :Inception3, :Inception4,
+          :SqueezeNet, :DenseNet, :MobileNetv1, :MobileNetv2, :MobileNetv3,
+          :MLPMixer, :ResMLP, :gMLP, :ViT, :ConvMixer, :ConvNeXt)
     @eval Base.show(io::IO, ::MIME"text/plain", model::$T) = _maybe_big_show(io, model)
 end
 

src/convnets/densenet.jl

@@ -11,11 +11,12 @@ Create a Densenet bottleneck layer
     (and scaling factor for inner feature maps; see ref)
 """
 function dense_bottleneck(inplanes, outplanes)
-  inner_channels = 4 * outplanes
-
-  SkipConnection(Chain(conv_bn((1, 1), inplanes, inner_channels; bias = false, rev = true)...,
-                       conv_bn((3, 3), inner_channels, outplanes; pad = 1, bias = false, rev = true)...),
-                 cat_channels)
+    inner_channels = 4 * outplanes
+    return SkipConnection(Chain(conv_bn((1, 1), inplanes, inner_channels; bias = false,
+                                        rev = true)...,
+                                conv_bn((3, 3), inner_channels, outplanes; pad = 1,
+                                        bias = false, rev = true)...),
+                          cat_channels)
 end
 
 """