Skip to content

Add rknn scripts#2012

Closed
boomermath wants to merge 17 commits intoPhotonVision:mainfrom
boomermath:main
Closed

Add rknn scripts#2012
boomermath wants to merge 17 commits intoPhotonVision:mainfrom
boomermath:main

Conversation

@boomermath
Copy link
Copy Markdown
Contributor

Description

Added multiple scripts for rknn conversion.

Meta

Merge checklist:

  • Pull Request title is short, imperative summary of proposed changes
  • The description documents the what and why
  • If this PR changes behavior or adds a feature, user documentation is updated
  • If this PR touches photon-serde, all messages have been regenerated and hashes have not changed unexpectedly
  • If this PR touches configuration, this is backwards compatible with settings back to v2024.3.1
  • If this PR touches pipeline settings or anything related to data exchange, the frontend typing is updated
  • If this PR addresses a bug, a regression test for it is added

@boomermath boomermath requested a review from a team as a code owner July 23, 2025 18:14
Copy link
Copy Markdown
Member

@samfreund samfreund left a comment

Choose a reason for hiding this comment

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

Running pip install...
ModuleNotFoundError                       Traceback (most recent call last)
[/content/create_onnx.py](https://localhost:8080/#) in <module>
     87 
     88     check_git_installed()
---> 89     run_onnx_conversion(args.version, args.model_path)

[/content/create_onnx.py](https://localhost:8080/#) in run_onnx_conversion(version, model_path)
     58         sys.exit(1)
     59 
---> 60     from ultralytics import YOLO
     61 
     62     model = YOLO(model_path)

ModuleNotFoundError: No module named 'ultralytics'

got this error running the notebook on Colab. I'm assuming you didn't pip install properly perhaps? Not sure. I'd also appreciate a progress bar for said pip install, if that's possible. If not, don't spend too much time on the progress bar.

Comment thread scripts/rknn-convert-tool/MAINTAINERS.md Outdated
Comment thread scripts/rknn-convert-tool/rknn_conversion.ipynb Outdated
Comment thread scripts/rknn-convert-tool/rknn_conversion.ipynb Outdated
Comment thread scripts/rknn-convert-tool/rknn_conversion.ipynb Outdated
Comment thread scripts/rknn-convert-tool/rknn_conversion.ipynb Outdated
Copy link
Copy Markdown
Member

@samfreund samfreund left a comment

Choose a reason for hiding this comment

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

Looks good so far, testing now

Comment thread scripts/rknn-convert-tool/rknn_conversion.ipynb Outdated
@boomermath boomermath closed this Aug 3, 2025
@boomermath
Copy link
Copy Markdown
Contributor Author

Will remake with correct branch

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants