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 and W should be divisible by stride and C >= (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>