ai.djl.basicmodelzoo.cv.classification

Class LeNet

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.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 and Description
static class	LeNet.Builder The Builder to construct a LeNet object.

Method Summary

Modifier and Type	Method and 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