mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-23 00:06:36 +00:00
zxsanny
20 lines
525 B
Bash
Executable File
20 lines
525 B
Bash
Executable File
# Use converter 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
|
|
|
|
# Use converter ONNX to RKNN
|
|
|
|
source ~/miniconda/bin/activate
|
|
conda activate toolkit2
|
|
cd rknn_model_zoo/examples/yolov8/python
|
|
python convert.py ../../../../azaion.onnx rk3588 i8
|
|
cp --verbose ../model/yolov8.rknn /azaion/models/azaion.rknn
|
|
conda deactivate
|
|
conda deactivate |