Uses of Class
org.deeplearning4j.nn.conf.layers.objdetect.Yolo2OutputLayer.Builder

Packages that use Yolo2OutputLayer.Builder

Package	Description
org.deeplearning4j.nn.conf.layers.objdetect

Uses of Yolo2OutputLayer.Builder in org.deeplearning4j.nn.conf.layers.objdetect

Methods in org.deeplearning4j.nn.conf.layers.objdetect that return Yolo2OutputLayer.Builder

Modifier and Type	Method and Description
Yolo2OutputLayer.Builder	Yolo2OutputLayer.Builder.boundingBoxPriors(org.nd4j.linalg.api.ndarray.INDArray boundingBoxes) Bounding box priors dimensions [width, height].
Yolo2OutputLayer.Builder	Yolo2OutputLayer.Builder.lambbaNoObj(double lambdaNoObj) Loss function coefficient for the "no object confidence" components of the loss function.
Yolo2OutputLayer.Builder	Yolo2OutputLayer.Builder.lambdaCoord(double lambdaCoord) Loss function coefficient for position and size/scale components of the loss function.
Yolo2OutputLayer.Builder	Yolo2OutputLayer.Builder.lossClassPredictions(org.nd4j.linalg.lossfunctions.ILossFunction lossClassPredictions) Loss function for the class predictions - defaults to L2 loss (i.e., sum of squared errors, as per the paper), however Loss MCXENT could also be used (which is more common for classification).
Yolo2OutputLayer.Builder	Yolo2OutputLayer.Builder.lossPositionScale(org.nd4j.linalg.lossfunctions.ILossFunction lossPositionScale) Loss function for position/scale component of the loss function