yolo3_grid_loss: Calculates loss for one 'Yolo3' grid.

Description Usage Arguments Value

View source: R/yolo3_loss.R

Description

Calculates loss for one 'Yolo3' grid.

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
yolo3_grid_loss(
  y_true,
  y_pred,
  anchors,
  n_class,
  nonobj_threshold,
  bbox_lambda = 1,
  obj_lambda = 1,
  noobj_lambda = 1,
  class_lambda = 1,
  class_weights = rep(1, n_class)
)

Arguments

y_true

Tensor of true coordinates/scores.

y_pred

Tensor of predicted coordinates/scores.

anchors

Prediction anchors (for one grid). For exact format check coco_anchors.

n_class

Number of prediction classes.

nonobj_threshold

Non-object ignore threshold.

bbox_lambda

Bounding box loss lambda.

obj_lambda

Object loss lambda.

noobj_lambda

Nonobject loss lambda.

class_lambda

Class loss lambda.

class_weights

Vector of length 'n_class' with class weights.

Value

Loss for one 'Yolo3' grid.


maju116/platypus documentation built on Oct. 18, 2020, 9:40 a.m.