generate_anchors: Calculates anchor boxes using k-mean++ algorithm.

Description Usage Arguments Value

View source: R/yolo3_anchors.R

Description

Calculates anchor boxes using k-mean++ algorithm.

Usage

1
2
3
4
5
6
7
8
9
generate_anchors(
  anchors_per_grid,
  annot_path,
  labels,
  n_iter = 10,
  annot_format = "pascal_voc",
  seed = 1234,
  centroid_fun = mean
)

Arguments

anchors_per_grid

Number of anchors per one grid.

annot_path

Annotations directory.

labels

Character vector containing class labels. For example coco_labels.

n_iter

Maximum number of iteration for k-mean++ algorithm.

annot_format

Annotations format. One of 'pascal_voc', 'labelme'.

seed

Random seed.

centroid_fun

Function to use for centroid calculation.

Value

List of anchor boxes.


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