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