add rknn conversion - install and use scripts, auto convert to rknn after AI training is done and put pt and rknn models to /azaion/models directory

orangepi5/convert-install.sh

@@ -29,5 +29,8 @@ 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

orangepi5/convert.sh

@@ -1,4 +1,4 @@
-# Use converter PT to ONNX
+# PT to ONNX
 cd rknn-convert/ultralytics_yolov8/
 cp --verbose /azaion/models/azaion.pt .
 source env/bin/activate
@@ -8,13 +8,12 @@ python ./ultralytics/engine/exporter.py
 cp --verbose azaion.onnx ../
 cd ..
 deactivate
+cp --verbose azaion.onnx /azaion/models/
 
-# Use converter ONNX to RKNN
-
+#  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
+python convert.py ../../../../azaion.onnx rk3588 i8 /azaion/models/azaion.rknn
 conda deactivate
 conda deactivate