org.deeplearning4j.zoo.model.helper

Class DarknetHelper

java.lang.Object

org.deeplearning4j.zoo.model.helper.DarknetHelper

public class DarknetHelper
extends Object

Contains functionality shared by Darknet19, TinyYOLO, and YOLO2.

Author:

saudet

Constructor Summary

Constructors

Constructor and Description
DarknetHelper()

Method Summary

Modifier and Type	Method and Description
static ComputationGraphConfiguration.GraphBuilder	addLayers(ComputationGraphConfiguration.GraphBuilder graphBuilder, int layerNumber, int filterSize, int nIn, int nOut, int poolSize)
static ComputationGraphConfiguration.GraphBuilder	addLayers(ComputationGraphConfiguration.GraphBuilder graphBuilder, int layerNumber, int filterSize, int nIn, int nOut, int poolSize, int poolStride)
static ComputationGraphConfiguration.GraphBuilder	addLayers(ComputationGraphConfiguration.GraphBuilder graphBuilder, int layerNumber, String input, int filterSize, int nIn, int nOut, int poolSize, int poolStride)
static int	getGridHeight(int[] inputShape) Returns inputShape[2] / 32, where inputShape[2] should be a multiple of 32.
static int	getGridWidth(int[] inputShape) Returns inputShape[1] / 32, where inputShape[1] should be a multiple of 32.

Methods inherited from class java.lang.Object

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

Constructor Detail

DarknetHelper

public DarknetHelper()

Method Detail

getGridWidth

public static int getGridWidth(int[] inputShape)

Returns inputShape[1] / 32, where inputShape[1] should be a multiple of 32.

getGridHeight

public static int getGridHeight(int[] inputShape)

Returns inputShape[2] / 32, where inputShape[2] should be a multiple of 32.

addLayers

public static ComputationGraphConfiguration.GraphBuilder addLayers(ComputationGraphConfiguration.GraphBuilder graphBuilder,
                                                                   int layerNumber,
                                                                   int filterSize,
                                                                   int nIn,
                                                                   int nOut,
                                                                   int poolSize)

addLayers

public static ComputationGraphConfiguration.GraphBuilder addLayers(ComputationGraphConfiguration.GraphBuilder graphBuilder,
                                                                   int layerNumber,
                                                                   int filterSize,
                                                                   int nIn,
                                                                   int nOut,
                                                                   int poolSize,
                                                                   int poolStride)

addLayers

public static ComputationGraphConfiguration.GraphBuilder addLayers(ComputationGraphConfiguration.GraphBuilder graphBuilder,
                                                                   int layerNumber,
                                                                   String input,
                                                                   int filterSize,
                                                                   int nIn,
                                                                   int nOut,
                                                                   int poolSize,
                                                                   int poolStride)