Package org.apache.camel.component.djl.model.cv

Class ZooImageGenerationPredictor

java.lang.Object

org.apache.camel.component.djl.model.AbstractPredictor

org.apache.camel.component.djl.model.cv.ZooImageGenerationPredictor

public class ZooImageGenerationPredictor
extends AbstractPredictor

Field Summary

Fields

Modifier and Type	Field	Description
protected ai.djl.repository.zoo.ZooModel<int[],ai.djl.modality.cv.Image[]>	model

Constructor Summary

Constructors

Constructor	Description
ZooImageGenerationPredictor(DJLEndpoint endpoint)

Method Summary

Modifier and Type	Method	Description
protected ai.djl.modality.cv.Image[]	predict(org.apache.camel.Exchange exchange, int[] seed)
void	process(org.apache.camel.Exchange exchange)

Methods inherited from class org.apache.camel.component.djl.model.AbstractPredictor

getEndpoint

Methods inherited from class java.lang.Object

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

Field Details

model

protected ai.djl.repository.zoo.ZooModel<int[],ai.djl.modality.cv.Image[]> model

Constructor Details

ZooImageGenerationPredictor

public ZooImageGenerationPredictor(DJLEndpoint endpoint)
                            throws ai.djl.repository.zoo.ModelNotFoundException,
ai.djl.MalformedModelException,
IOException

Throws:

ai.djl.repository.zoo.ModelNotFoundException

ai.djl.MalformedModelException

IOException

Method Details

process

public void process(org.apache.camel.Exchange exchange)

Specified by:

process in class AbstractPredictor

predict

protected ai.djl.modality.cv.Image[] predict(org.apache.camel.Exchange exchange,
 int[] seed)