ReorgYolo Layer#
Versioned name: ReorgYolo-1
Category: Object detection
Short description: ReorgYolo reorganizes input tensor taking into account strides.
Detailed description: Reference
Attributes
stride
Description: stride is the distance between cut throws in output blobs.
Range of values: positive integer
Type:
int
Required: yes
Inputs:
1: 4D input tensor of any type and shape
[N, C, H, W]
.H
andW
should be divisible bystride
andC >= (stride*stride)
. Required.
Outputs:
1: 4D output tensor of the same type as input tensor and shape
[N, C*stride*stride, H/stride, W/stride]
.
Example
<layer id="89" name="reorg" type="ReorgYolo">
<data stride="2"/>
<input>
<port id="0">
<dim>1</dim>
<dim>64</dim>
<dim>26</dim>
<dim>26</dim>
</port>
</input>
<output>
<port id="1" precision="f32">
<dim>1</dim>
<dim>256</dim>
<dim>13</dim>
<dim>13</dim>
</port>
</output>
</layer>