Skip to content

Layout Support#13

Merged
kozlov721 merged 14 commits intomainfrom
feature/layout-support
Jul 25, 2024
Merged

Layout Support#13
kozlov721 merged 14 commits intomainfrom
feature/layout-support

Conversation

@kozlov721
Copy link
Copy Markdown
Collaborator

@kozlov721 kozlov721 commented Jul 24, 2024

  • Added support for layout field in the config for both inputs and outputs
  • Output shape is always read from the converted model
    • Output layout is inferred based on the output shape to detect when there's a switch from "NCHW" to "NHWC" convention on some platforms
  • hailo.early_stop renamed to hailo.skip_compilation

Example

Input to ONNX in "NCHW":

inputs:
  - name: "input"
    shape: [1, 3, 256, 256]
    layout: "NCHW"

After conversion to RVC4, the generated NN Archive will contain:

"inputs": [
    {
        "name": "input",
        "shape": [1, 256, 256, 3],
        "layout": "NHWC",
    }
]

@kozlov721 kozlov721 added the enhancement New feature or request label Jul 24, 2024
@kozlov721 kozlov721 self-assigned this Jul 24, 2024
@kozlov721 kozlov721 added RVC2 Changes affecting RVC2 export RVC3 Changes affecting RVC3 export RVC4 Changes affecting RVC4 export Hailo Changes affecting Hailo export NN Archive Changes affecting NN Archive export labels Jul 25, 2024
@kozlov721 kozlov721 merged commit 895499f into main Jul 25, 2024
@kozlov721 kozlov721 deleted the feature/layout-support branch July 25, 2024 17:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request Hailo Changes affecting Hailo export NN Archive Changes affecting NN Archive export RVC2 Changes affecting RVC2 export RVC3 Changes affecting RVC3 export RVC4 Changes affecting RVC4 export

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant