xref: /OK3568_Linux_fs/external/rockchip-test/npu2/rknn_demo.sh (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun#!/bin/bash
2*4882a593Smuzhiyun
3*4882a593SmuzhiyunCOMPATIBLE=$(cat /proc/device-tree/compatible)
4*4882a593Smuzhiyunif [[ $(expr $COMPATIBLE : ".*rk3588") -ne 0 ]]; then
5*4882a593Smuzhiyun	rknn_common_test /usr/share/model/RK3588/mobilenet_v1.rknn /usr/share/model/dog_224x224.jpg 10
6*4882a593Smuzhiyunelif [[ $(expr $COMPATIBLE : ".*rk3568") -ne 0 ]]; then
7*4882a593Smuzhiyun	rknn_common_test /usr/share/model/RK3566_RK3568/mobilenet_v1.rknn /usr/share/model/dog_224x224.jpg 10
8*4882a593Smuzhiyunelif [[ $(expr $COMPATIBLE : ".*rk3566") -ne 0 ]]; then
9*4882a593Smuzhiyun	rknn_common_test /usr/share/model/RK3566_RK3568/mobilenet_v1.rknn /usr/share/model/dog_224x224.jpg 10
10*4882a593Smuzhiyunelif [[ $(expr $COMPATIBLE : ".*rk3562") -ne 0 ]]; then
11*4882a593Smuzhiyun	rknn_common_test /usr/share/model/RK3562/mobilenet_v1.rknn /usr/share/model/dog_224x224.jpg 10
12*4882a593Smuzhiyunelse
13*4882a593Smuzhiyun	echo "The RKNPU2 did't support this Socs yet..."
14*4882a593Smuzhiyunfi
15