RK3399pro Rock-X python代码+USB摄像头目标检测报错

KevinWu

平台：Ubuntu 20.04 + 2GB内存 +1GB NPU

问题1：按照Rockchip_Developer_Guide_RockX_SDK_CN 文档，接入USB摄像头后，运行以下命令

1. python3 -m rockx.test.camera.rockx_object_detection

复制代码

1. python3 -m rockx_object_detection.py

复制代码

报以下错误：内存不够

1. Traceback (most recent call last):
   
2.   File "/usr/lib/python3.8/runpy.py", line 194, in _run_module_as_main
   
3.     return _run_code(code, main_globals, None,
   
4.   File "/usr/lib/python3.8/runpy.py", line 87, in _run_code
   
5.     exec(code, run_globals)
   
6.   File "/home/firefly/.local/lib/python3.8/site-packages/rockx/test/camera/rockx_object_detection.pine 12, in <module>
   
7.     object_det_handle = RockX(RockX.ROCKX_MODULE_OBJECT_DETECTION, target_device=args.device)
   
8.   File "/home/firefly/.local/lib/python3.8/site-packages/rockx/RockX.py", line 235, in __init__
   
9.     self.rockx_base = RockXBase()
   
10.   File "/home/firefly/.local/lib/python3.8/site-packages/rockx/RockXBase.py", line 218, in __init__
    
11.     RockXBase._load_library()
    
12.   File "/home/firefly/.local/lib/python3.8/site-packages/rockx/RockXBase.py", line 448, in _load_li
    
13.     RockXBase.lib = CDLL(os.path.join(lib_dir, 'librockx.so'))
    
14.   File "/usr/lib/python3.8/ctypes/__init__.py", line 373, in __init__
    
15.     self._handle = _dlopen(self._name, mode)
    
16. 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权限运行

1. sudo python3 -m rockx.test.camera.rockx_object_detection

复制代码

1. sudo python3 -m rockx_object_detection.py

复制代码

报以下错误：
ModuleNotFoundError: No module named 'rockx'




3.参考文档和代码

1. import argparse
   
2. from rockx import RockX
   
3. import cv2
   
4. 
   
5. if __name__ == '__main__':
   
6. 
   
7.     parser = argparse.ArgumentParser(description="RockX Object Detection Demo")
   
8.     parser.add_argument('-c', '--camera', help="camera index", type=int, default=0)
   
9.     parser.add_argument('-d', '--device', help="target device id", type=str)
   
10.     args = parser.parse_args()
    
11. 
    
12.     object_det_handle = RockX(RockX.ROCKX_MODULE_OBJECT_DETECTION, target_device=args.device)
    
13. 
    
14.     cap = cv2.VideoCapture(args.camera)
    
15.     cap.set(3, 1280)
    
16.     cap.set(4, 720)
    
17.     last_face_feature = None
    
18. 
    
19.     while True:
    
20.         # Capture frame-by-frame
    
21.         ret, frame = cap.read()
    
22. 
    
23.         in_img_h, in_img_w = frame.shape[:2]
    
24. 
    
25.         ret, results = object_det_handle.rockx_face_detect(frame, in_img_w, in_img_h, RockX.ROCKX_PIXEL_FORMAT_BGR888)
    
26. 
    
27.         for result in results:
    
28.             cv2.rectangle(frame,
    
29.                           (result.box.left, result.box.top),
    
30.                           (result.box.right, result.box.bottom),
    
31.                           (0, 255, 0), 2)
    
32.             cv2.putText(frame, "%s" % RockX.ROCKX_OBJECT_DETECTION_LABELS_91[result.cls_idx],
    
33.                         (result.box.left, result.box.top - 10),
    
34.                         cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0))
    
35. 
    
36.         # Display the resulting frame
    
37.         cv2.imshow('RockX Object Detection - ' + str(args.device), frame)
    
38.         if cv2.waitKey(1) & 0xFF == ord('q'):
    
39.             break
    
40. 
    
41.     # When everything done, release the capture
    
42.     cap.release()
    
43.     cv2.destroyAllWindows()
    
44. 
    
45.     object_det_handle.release()
    

复制代码

neutionwei

你好，先确认一下有没有安装rknn相关包。

KevinWu

我确实也是按照他们的文档操作，安装对应的包【RockX-1.4.4-py3-none-any.whl】，好像也不行


我在终端输入Python3，然后输入 from rockx import RockX 是没问题的 ；在线求助
   

KevinWu

neutionwei 发表于 2023-4-25 14:35
你好，先确认一下有没有安装rknn相关包。

我之前有用你们的demo 来做单张图片推理，是可以的，你可以看看我楼下的评论

KevinWu

我之前有用官方提供的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， 麻烦你帮我确认一下，感谢！

neutionwei

之前OK的系统版本与现在的是一样嘛？

KevinWu

neutionwei 发表于 2023-4-25 15:17
之前OK的系统版本与现在的是一样嘛？

是的，一样的主板RK3399 PRO  （2GB内存+1GB NPU），环境也是一样的

板蓝根

你的 rockx 是装在哪个用户上的，一时sudo 一时普通用户的，我看你的普通 python3 调试器都可以 import rockx 了

neutionwei

环境一样，按道理是没问题的，你看看是不是哪里改动了。

KevinWu

板蓝根 发表于 2023-4-25 17:20
你的 rockx 是装在哪个用户上的，一时sudo 一时普通用户的，我看你的普通 python3 调试器都可以 import roc ...

我一直都是在firefly 用户操作，之前也是按照操作指示Opencv 编译和安装编译成功了opencv 3.4.15，然后接入USB摄像头是正常的，现在就是想用python代码来进行目标检测，一开始执行命令，说内存不够，无法正常运行，后面就是sudo权限可以执行，但是就是没有RockX模块。我等会尝试一下创建python3.7虚拟环境再测试，感谢指示