Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Layout Support #13

Merged
merged 14 commits into from
Jul 25, 2024
Merged

Layout Support #13

merged 14 commits into from
Jul 25, 2024

Conversation

kozlov721
Copy link
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
6 of 7 checks passed
@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