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>