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

CLI Archive Command #20

Merged
merged 3 commits into from
Aug 5, 2024
Merged

CLI Archive Command #20

merged 3 commits into from
Aug 5, 2024

Conversation

kozlov721
Copy link
Collaborator

@kozlov721 kozlov721 commented Jul 25, 2024

Added an archive command to the CLI. This command takes a model file and creates an NN Archive.
This command runs outside of the docker container.

Currently supported model formats:

  • ONNX
  • tflite
  • IR
  • dlc (if snpe-dlc-info is in path)
  • har (if hailo_sdk_client is installed in the python environment)

Usage:

modelconverter archive path/to/model_file --save-path gs://remote/path --put-file-plugin plugin 

@kozlov721 kozlov721 added enhancement New feature or request CLI Changes affecting the CLI labels Jul 25, 2024
@kozlov721 kozlov721 self-assigned this Jul 25, 2024
modelconverter/__main__.py Show resolved Hide resolved
"name": name,
"shape": shape,
"layout": make_default_layout(shape),
"dtype": "float32",
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Couldn't it be possible to have int8 with dlc? Maybe also tflite? Would there be some way to detect that?

Comment on lines +248 to +251
"mean": None,
"scale": None,
"reverse_channels": False,
"interleaved_to_planar": False,
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is there some way to detect these? Is it safe to assume these "defaults"?

@kozlov721 kozlov721 merged commit 6468043 into main Aug 5, 2024
1 check passed
@kozlov721 kozlov721 deleted the feature/archive-generator branch August 5, 2024 17:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLI Changes affecting the CLI enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants