mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-23 05:06:36 +00:00
zxsanny
19 lines
504 B
Bash
Executable File
19 lines
504 B
Bash
Executable File
# PT to ONNX
|
|
cd rknn-convert/ultralytics_yolov8/
|
|
cp --verbose /azaion/models/azaion.pt .
|
|
source env/bin/activate
|
|
pip install onnx
|
|
export PYTHONPATH=./
|
|
python ./ultralytics/engine/exporter.py
|
|
cp --verbose azaion.onnx ../
|
|
cd ..
|
|
deactivate
|
|
cp --verbose azaion.onnx /azaion/models/
|
|
|
|
# ONNX to RKNN
|
|
source ~/miniconda/bin/activate
|
|
conda activate toolkit2
|
|
cd rknn_model_zoo/examples/yolov8/python
|
|
python convert.py ../../../../azaion.onnx rk3588 i8 /azaion/models/azaion.rknn
|
|
conda deactivate
|
|
conda deactivate |