MixFormerV2 は、改良型の統合Transformerターゲット追跡モデルで、特徴抽出とマッチングプロセスを融合することで、追跡の精度と速度を向上させます。
git clone https://huggingface.co/AXERA-TECH/MixFormerV2
ファイル説明:
m5stack@raspberrypi:~/rsp/MixFormerV2 $ ls -lh
total 63M
drwxrwxr-x 2 m5stack m5stack 4.0K Aug 11 18:28 ax630c
drwxrwxr-x 2 m5stack m5stack 4.0K Aug 11 18:28 ax650
-rw-rw-r-- 1 m5stack m5stack 63M Aug 11 18:28 car.avi
-rw-rw-r-- 1 m5stack m5stack 0 Aug 11 18:28 config.json
drwxrwxr-x 2 m5stack m5stack 4.0K Aug 11 18:28 onnx
-rw-rw-r-- 1 m5stack m5stack 4.0K Aug 11 18:28 README.md
-rw-rw-r-- 1 m5stack m5stack 15K Aug 11 18:28 run_mixformer2_axmodel.py
-rw-rw-r-- 1 m5stack m5stack 14K Aug 11 18:28 run_mixformer2_onnx.py
python -m venv mixformer
source mixformer/bin/activate
pip install https://github.com/AXERA-TECH/pyaxengine/releases/download/0.1.3.rc1/axengine-0.1.3-py3-none-any.whl
pip install argparse numpy opencv-python glob2
python3 run_mixformer2_axmodel.py --model-path ax650/mixformer_v2.axmodel --frame-path car.avi -r 10
実行結果:
(mixformer) m5stack@raspberrypi:~/rsp/MixFormerV2 $ python3 run_mixformer2_axmodel.py --model-path ax650/mixformer_v2.axmodel --frame-path car.avi -r 10
[INFO] Available providers: ['AXCLRTExecutionProvider']
[INFO] Using provider: AXCLRTExecutionProvider
[INFO] SOC Name: AX650N
[INFO] VNPU type: VNPUType.DISABLED
[INFO] Compiler version: 3.4-dirty 4ff37520-dirty
====================type================= [1079, 482] <class 'list'> <class 'list'>
最初のフレームの初期化が完了しました!
Video: tracking 1011.0fps
Video: tracking 8.0fps
Video: tracking 8.0fps
Video: tracking 8.0fps
Video: tracking 8.0fps
Video: tracking 8.0fps
Video: tracking 8.0fps
Video: tracking 8.0fps
Video: tracking 8.0fps
Video: tracking 8.0fps
Video: tracking 8.0fps
最大フレーム数 (10) に達しました。ループを終了します。
video: 平均最終トラッキング fps 121.2 fps