RK3399pro Rock-X python代码+USB摄像头目标检测报错
本帖最后由 KevinWu 于 2023-4-25 09:20 编辑平台：Ubuntu 20.04 + 2GB内存 +1GB NPU
问题1：按照Rockchip_Developer_Guide_RockX_SDK_CN 文档，接入USB摄像头后，运行以下命令python3 -m rockx.test.camera.rockx_object_detectionpython3 -m rockx_object_detection.py报以下错误：内存不够
Traceback (most recent call last):
File "/usr/lib/python3.8/runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/usr/lib/python3.8/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/home/firefly/.local/lib/python3.8/site-packages/rockx/test/camera/rockx_object_detection.pine 12, in <module>
object_det_handle = RockX(RockX.ROCKX_MODULE_OBJECT_DETECTION, target_device=args.device)
File "/home/firefly/.local/lib/python3.8/site-packages/rockx/RockX.py", line 235, in __init__
self.rockx_base = RockXBase()
File "/home/firefly/.local/lib/python3.8/site-packages/rockx/RockXBase.py", line 218, in __init__
RockXBase._load_library()
File "/home/firefly/.local/lib/python3.8/site-packages/rockx/RockXBase.py", line 448, in _load_li
RockXBase.lib = CDLL(os.path.join(lib_dir, 'librockx.so'))
File "/usr/lib/python3.8/ctypes/__init__.py", line 373, in __init__
self._handle = _dlopen(self._name, mode)
OSError: /home/firefly/.local/lib/python3.8/site-packages/rockx/sdk/rockx-rk3399pro-Linux/lib64/lib.so: cannot allocate memory in static TLS block
问题2：通过chat-gpt搜索了解决答案，用sudo权限运行
sudo python3 -m rockx.test.camera.rockx_object_detectionsudo python3 -m rockx_object_detection.py报以下错误：
ModuleNotFoundError: No module named 'rockx'
3.参考文档和代码
import argparse
from rockx import RockX
import cv2
if __name__ == '__main__':
parser = argparse.ArgumentParser(description="RockX Object Detection Demo")
parser.add_argument('-c', '--camera', help="camera index", type=int, default=0)
parser.add_argument('-d', '--device', help="target device id", type=str)
args = parser.parse_args()
object_det_handle = RockX(RockX.ROCKX_MODULE_OBJECT_DETECTION, target_device=args.device)
cap = cv2.VideoCapture(args.camera)
cap.set(3, 1280)
cap.set(4, 720)
last_face_feature = None
while True:
# Capture frame-by-frame
ret, frame = cap.read()
in_img_h, in_img_w = frame.shape[:2]
ret, results = object_det_handle.rockx_face_detect(frame, in_img_w, in_img_h, RockX.ROCKX_PIXEL_FORMAT_BGR888)
for result in results:
cv2.rectangle(frame,
(result.box.left, result.box.top),
(result.box.right, result.box.bottom),
(0, 255, 0), 2)
cv2.putText(frame, "%s" % RockX.ROCKX_OBJECT_DETECTION_LABELS_91,
(result.box.left, result.box.top - 10),
cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0))
# Display the resulting frame
cv2.imshow('RockX Object Detection - ' + str(args.device), frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# When everything done, release the capture
cap.release()
cv2.destroyAllWindows()
object_det_handle.release()
你好，先确认一下有没有安装rknn相关包。 本帖最后由 KevinWu 于 2023-4-25 14:45 编辑
我确实也是按照他们的文档操作，安装对应的包【RockX-1.4.4-py3-none-any.whl】，好像也不行
我在终端输入Python3，然后输入 from rockx import RockX 是没问题的 ；在线求助
neutionwei 发表于 2023-4-25 14:35
你好，先确认一下有没有安装rknn相关包。
我之前有用你们的demo 来做单张图片推理，是可以的，你可以看看我楼下的评论
本帖最后由 KevinWu 于 2023-4-25 15:06 编辑
我之前有用官方提供的demo
Rock-X/RockX_SDK_V1.4.4_20210623/demo/command_line_demo/rockx_object_detection_demo/rockx_object_detection.cpp来做单张图片推理，是有安装RKNN 驱动，驱动版本是1.7.1， 麻烦你帮我确认一下，感谢！
之前OK的系统版本与现在的是一样嘛？ neutionwei 发表于 2023-4-25 15:17
之前OK的系统版本与现在的是一样嘛？
是的，一样的主板RK3399 PRO（2GB内存+1GB NPU），环境也是一样的 你的 rockx 是装在哪个用户上的，一时sudo 一时普通用户的，我看你的普通 python3 调试器都可以 import rockx 了 环境一样，按道理是没问题的，你看看是不是哪里改动了。 板蓝根 发表于 2023-4-25 17:20
你的 rockx 是装在哪个用户上的，一时sudo 一时普通用户的，我看你的普通 python3 调试器都可以 import roc ...
我一直都是在firefly 用户操作，之前也是按照操作指示Opencv 编译和安装编译成功了opencv 3.4.15，然后接入USB摄像头是正常的，现在就是想用python代码来进行目标检测，一开始执行命令，说内存不够，无法正常运行，后面就是sudo权限可以执行，但是就是没有RockX模块。我等会尝试一下创建python3.7虚拟环境再测试，感谢指示
页:
[1]
2