diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml index 0a4d4e523..1699a2011 100644 --- a/.github/workflows/ci-testing.yml +++ b/.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 diff --git a/data/rknpu-qnt.txt b/data/rknpu-qnt.txt new file mode 100644 index 000000000..46c346033 --- /dev/null +++ b/data/rknpu-qnt.txt @@ -0,0 +1,2 @@ +images/bus.jpg +images/zidane.jpg diff --git a/export.py b/export.py index 20c739be0..b973bf79d 100644 --- a/export.py +++ b/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