Package ai.djl.basicmodelzoo.cv.classification

Class LeNet

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.LeNet

public final class LeNet
extends 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 final class	LeNet.Builder	The Builder to construct a LeNet object.

Method Summary

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 Details

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