Package ai.djl.basicmodelzoo.cv.classification

Class LeNet

  • public final class LeNet
extends java.lang.Object
    The model was introduced (and named for) Yann Lecun, for the purpose of recognizing handwritten digits in images [LeNet5](http://yann.lecun.com/exdb/lenet/).
    See Also:
    The D2L chapter on LeNet

    • Nested Class Summary

      Nested Classes 
      Modifier and Type Class Description
      static class  LeNet.Builder
      The Builder to construct a LeNet object.

    • Method Summary

      All Methods Static Methods Concrete Methods 
      Modifier and Type Method Description
      static LeNet.Builder builder()
      Creates a builder to build a LeNet.
      static ai.djl.nn.Block leNet​(LeNet.Builder builder)
      Creates a LeNet network block with the help of the LeNet Builder.

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

    • Method Detail

      • leNet

        public static ai.djl.nn.Block leNet​(LeNet.Builder builder)
        Creates a LeNet network block with the help of the LeNet Builder.
        Parameters:
        builder - the LeNet.Builder with the necessary arguments.
        Returns:
        a LeNet block.

      • builder

        public static LeNet.Builder builder()
        Creates a builder to build a LeNet.
        Returns:
        a new builder