Sign inSign up

firesh/rknn-toolkit2

By firesh

•Updated about 2 years ago

Image
Internet of things
Machine learning & AI
Developer tools
1

3.0K

firesh/rknn-toolkit2 repository overview

⁠rknn-toolkit2

该 Docker 镜像系列基于 Python，不同的标签代表了不同的 Python 版本

⁠tags

  • cp311: Python 3.11
  • cp310: Python 3.10
  • cp39: Python 3.9
  • cp38: Python 3.8

⁠Dockerfile

Dockerfiles⁠

⁠运行示例

⁠使用 cp311 标签的示例
docker pull firesh/rknn-toolkit2:cp311
docker run -it firesh/rknn-toolkit2:cp311 python --version
⁠使用model_zoo
cd /workspaces
git clone https://github.com/airockchip/rknn_model_zoo zoo
docker run -it -v /workspaces/zoo:/zoo firesh/rknn-toolkit2:cp311 bash
cd /zoo/examples/yolov5/model/
# 下载模型
cd ../python
# 转换模型
python convert.py ../model/yolov5s.onnx rk3588
python convert.py ../model/yolov5s.onnx rk3588 fp ../model/yolov5s_fp.rknn

⁠Issues

https://github.com/firesh/rknn-toolkit2/issues⁠

Tag summary

Content type

Image

Digest

sha256:0e78e046e…

Size

5.2 GB

Last updated

about 2 years ago

docker pull firesh/rknn-toolkit2