- from rknnlite.api import RKNNLite
- import toybrick as toy
- import numpy as np
- def init_rknn(model):
- rknn_lite = RKNNLite()
- ret = rknn_lite.load_rknn(model)
- if ret != 0:
- print('Load RKNN model failed')
- exit(ret)
- print('--> Init runtime environment')
- ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0_1_2)
- if ret != 0:
- print('Init runtime environment failed')
- exit(ret)
- print('done')
- return rknn_lite
- if __name__ == '__main__':
- rknn = init_rknn('./model.rknn')
- hdmi = toy.HdmiCapture()
- disp = toy.Display('', 1920, 1080, False)
- print("Connect display width = " + str(disp.width) + " ; height = " + str(disp.height))
- while True:
- ret, frame = hdmi.read(1920, 1080, toy.FMT_RGB888)
- if frame is not None:
- img = frame.resize(320, 320) # 这里的img是一个dma buffer
- npimg = img.asarray() # 必须要单独再用一个变量,保证img变量引用计数不会为0,因为这里的asarray是0拷贝构造,如果img被析构,npimg指向的内存空间就是错的了
- outputs = rknn.inference(inputs=[npimg])
- print(outputs)
复制代码
mike841211 发表于 2023-3-21 08:44
为什么toybrick 不搞个面向通用的一个包呢?只能在toybrick的板上用
jefferyzhang 发表于 2023-3-21 08:47
对外发布的EdgeSDK里都包含该包,但我们是toybrick论坛,我们当然只维护toybrick板子,其他板子请联系板 ...
hiCrystal 发表于 2023-8-28 14:20
请问PythonSDK必须是板商在固件上支持吗?可以在刷好系统的板子自己安装PythonSDK吗? ...
jefferyzhang 发表于 2023-8-28 15:35
PythonSDK是开放免费用的,
但我们只对toybrick板子和edge sdk编写和测试,其他板子我们没办法顾及,
如 ...
hiCrystal 发表于 2023-8-29 11:19
请问PythonSDK的源代码开放吗?如果其他板子想要支持PythonSDK是不是只能在编译固件时就要操作? ...
欢迎光临 Toybrick (https://t.rock-chips.com/) | Powered by Discuz! X3.3 |