org.deeplearning4j.zoo.model

Class Darknet19

java.lang.Object

org.deeplearning4j.zoo.ZooModel

org.deeplearning4j.zoo.model.Darknet19

All Implemented Interfaces:

InstantiableModel

public class Darknet19
extends ZooModel

Darknet19
Reference: https://arxiv.org/pdf/1612.08242.pdf

ImageNet weights for this model are available and have been converted from https://pjreddie.com/darknet/imagenet/ using https://github.com/allanzelener/YAD2K .

There are 2 pretrained models, one for 224x224 images and one fine-tuned for 448x448 images. Call setInputShape() with either {3, 224, 224} or {3, 448, 448} before initialization. The channels of the input images need to be in RGB order (not BGR), with values normalized within [0, 1]. The output labels are as per https://github.com/pjreddie/darknet/blob/master/data/imagenet.shortnames.list .

Author:

saudet

Method Summary

Modifier and Type	Method and Description
ComputationGraphConfiguration	conf()
ComputationGraph	init()
ModelMetaData	metaData()
Class<? extends Model>	modelType()
long	pretrainedChecksum(PretrainedType pretrainedType)
String	pretrainedUrl(PretrainedType pretrainedType)
void	setInputShape(int[][] inputShape)

Methods inherited from class org.deeplearning4j.zoo.ZooModel

initPretrained, initPretrained, modelName, pretrainedAvailable

Methods inherited from class java.lang.Object

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

Method Detail

pretrainedUrl

public String pretrainedUrl(PretrainedType pretrainedType)

pretrainedChecksum

public long pretrainedChecksum(PretrainedType pretrainedType)

modelType

public Class<? extends Model> modelType()

conf

public ComputationGraphConfiguration conf()

init

public ComputationGraph init()

metaData

public ModelMetaData metaData()

setInputShape

public void setInputShape(int[][] inputShape)