ngraph.region_yolo

ngraph.region_yolo(input: _pyngraph.Node, coords: int, classes: int, num: int, do_softmax: bool, mask: List[int], axis: int, end_axis: int, anchors: Optional[List[float]] = None, name: Optional[str] = None) _pyngraph.Node

Return a node which produces the RegionYolo operation.

Parameters
  • input – Input data

  • coords – Number of coordinates for each region

  • classes – Number of classes for each region

  • num – Number of regions

  • do_softmax – Compute softmax

  • mask – Mask

  • axis – Axis to begin softmax on

  • end_axis – Axis to end softmax on

  • anchors – A flattened list of pairs [width, height] that describes prior box sizes

  • name – Optional name for output node.

Returns

RegionYolo node