Class MobileNetV1.Builder

  • Enclosing class:
    MobileNetV1

    public static final class MobileNetV1.Builder
    extends java.lang.Object
    The Builder to construct a MobileNetV1 object.
    • Method Detail

      • optWidthMultiplier

        public MobileNetV1.Builder optWidthMultiplier​(float widthMultiplier)
        Sets the widthMultiplier of MobileNet.
        Parameters:
        widthMultiplier - the widthMultiplier of MobileNet
        Returns:
        this Builder
      • optBatchNormMomentum

        public MobileNetV1.Builder optBatchNormMomentum​(float batchNormMomentum)
        Sets the momentum of batchNorm layer.
        Parameters:
        batchNormMomentum - the momentum
        Returns:
        this Builder
      • setOutSize

        public MobileNetV1.Builder setOutSize​(long outSize)
        Sets the size of the output.
        Parameters:
        outSize - the output size
        Returns:
        this Builder