Class MobileNetV1


  • public final class MobileNetV1
    extends java.lang.Object
    MobileNetV1 contains a generic implementation of Mobilenet adapted from https://github.com/weiaicunzai/pytorch-cifar100/blob/master/models/mobilenet.py (Original author weiaicunzai).

    see https://arxiv.org/pdf/1704.04861.pdf for more information about MobileNet

    • Method Detail

      • depthSeparableConv2d

        public static ai.djl.nn.Block depthSeparableConv2d​(int inputChannels,
                                                           int outputChannels,
                                                           int stride,
                                                           MobileNetV1.Builder builder)
        Builds a Block that represent a depthWise-pointWise Unit used in the implementation of the MobileNet Model.
        Parameters:
        inputChannels - number of inputChannels, used for depthWise Kernel
        outputChannels - number of outputChannels, used for pointWise kernel
        stride - control the stride of depthWise Kernel
        builder - add the builder to obtain batchNormMomentum
        Returns:
        a Block that represent a depthWise-pointWise Unit