faces_server_python_rknpu_o.../app/models/test_rknn.py
2025-12-27 15:36:20 +01:00

19 lines
404 B
Python

from rknn.api import RKNN
print("Testing RKNN model load...")
rknn = RKNN()
ret = rknn.load_rknn("models/rknn/scrfd.rknn")
if ret != 0:
print("❌ Failed to load RKNN model")
else:
print("✅ RKNN model loaded successfully")
print("Init runtime (CPU fallback)...")
ret = rknn.init_runtime()
if ret != 0:
print("❌ Runtime init failed")
else:
print("✅ Runtime init OK (CPU mode)")