Package smile.vision

Class VisionModel

java.lang.Object

smile.deep.Model

smile.vision.VisionModel

All Implemented Interfaces:

Function<Tensor,Tensor>

public class VisionModel
extends Model

The computer vision models.

Constructor Summary

Constructors

Constructor	Description
VisionModel(LayerBlock net, Transform transform)	Constructor.

Method Summary

Modifier and Type	Method	Description
Tensor	forward(BufferedImage... images)	Forward propagation (or forward pass) through the model.
Transform	transform()	Returns the associated image transform.

Methods inherited from class smile.deep.Model

apply, asTorch, device, eval, eval, forward, load, save, setLearningRateSchedule, to, toString, train, train, train

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.function.Function

andThen, compose

Constructor Details

VisionModel

public VisionModel(LayerBlock net,
 Transform transform)

Constructor.

Parameters:

net - the neural network.

transform - the image transform.

Method Details

forward

public Tensor forward(BufferedImage... images)

Forward propagation (or forward pass) through the model.

Parameters:

images - the input images.

Returns:

the output tensor.

transform

public Transform transform()

Returns the associated image transform.

Returns:

the associated image transform.