mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 22:36:36 +00:00
zxsanny
37 lines
1.2 KiB
Bash
37 lines
1.2 KiB
Bash
mkdir rknn-convert
|
|
cd rknn-convert
|
|
|
|
# Install converter PT to ONNX
|
|
git clone https://github.com/airockchip/ultralytics_yolov8
|
|
cd ultralytics_yolov8
|
|
sudo apt install python3.12-venv
|
|
python3 -m venv env
|
|
source env/bin/activate
|
|
pip install .
|
|
pip install onnx
|
|
cp ultralytics/cfg/default.yaml ultralytics/cfg/default_backup.yaml
|
|
sed -i -E "s/(model: ).+( #.+)/\1azaion.pt\2/" ultralytics/cfg/default.yaml
|
|
cd ..
|
|
deactivate
|
|
|
|
# Install converter ONNX to RKNN
|
|
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh
|
|
chmod +x miniconda.sh
|
|
bash miniconda.sh -b -p $HOME/miniconda
|
|
|
|
source ~/miniconda/bin/activate
|
|
conda create -n toolkit2 -y python=3.11
|
|
conda activate toolkit2
|
|
git clone https://github.com/rockchip-linux/rknn-toolkit2.git
|
|
cd rknn-toolkit2/rknn-toolkit2/packages
|
|
pip install -r requirements_cp311-1.6.0.txt
|
|
pip install rknn_toolkit2-1.6.0+81f21f4d-cp311-cp311-linux_x86_64.whl
|
|
pip install "numpy<2.0"
|
|
cd ../../../
|
|
git clone https://github.com/airockchip/rknn_model_zoo.git
|
|
|
|
sed -i -E "s#(DATASET_PATH = ').+(')#\1/azaion/data-sample/azaion_subset.txt\2 #" rknn_model_zoo/examples/yolov8/python/convert.py
|
|
|
|
conda deactivate
|
|
conda deactivate
|