fix rknpu export

.github/workflows/ci-testing.yml

@@ -73,4 +73,4 @@ jobs:
             done
           done
           python models/yolo.py --cfg $m.yaml  # build PyTorch model
-          python export.py --weights $m.pt --img 64  # export
+          python export.py --weights $m.pt --img 64 --rknpu --int8 --data data/rknpu-qnt.txt # export

data/rknpu-qnt.txt

@@ -0,0 +1,2 @@
+images/bus.jpg
+images/zidane.jpg

export.py

@@ -61,6 +61,8 @@ from pathlib import Path
 if '--rknpu' in sys.argv:
     os.environ['RKNN_model_hack'] = "1"
     rknpu = True
+else:
+    rknpu = False
 
 import pandas as pd
 import torch