indra4837 / yolov4_trt_ros

YOLOv4 object detector using TensorRT engine
MIT License
100 stars 31 forks source link

object has no attribute'allocate_buffers' #23

Open CYong030 opened 2 years ago

CYong030 commented 2 years ago

I met some problem and i can't solve it.

i have two solve direction. My python have error. My python version is 3.6.9 Cuda doesn't clear completely.

If i get some help, i will appreciate .

`PARAMETERS

NODES / yolov4_trt_node (yolov4_trt_ros/trt_yolo_v4.py)

auto-starting new master process[master]: started with pid [15199] ROS_MASTER_URI=http://localhost:11311

setting /run_id to 291d39c4-29d0-11ed-b07f-60a4b7c96821 process[rosout-1]: started with pid [15212] started core service [/rosout] process[yolov4_trt_node-2]: started with pid [15215] Traceback (most recent call last): File "/home/jetson/catkin_ws/src/yolov4_trt_ros/trt_yolo_v4.py", line 175, in main() File "/home/jetson/catkin_ws/src/yolov4_trt_ros/trt_yolo_v4.py", line 164, in main yolo = yolov4() File "/home/jetson/catkin_ws/src/yolov4_trt_ros/trt_yolo_v4.py", line 35, in init (self.model_path + self.model), (self.h, self.w), self.category_num) File "/home/jetson/catkin_ws/src/yolov4_trt_ros/utils/yolo_with_plugins.py", line 221, in init self.inputs, self.outputs, self.bindings, self.stream ,grid_sizes= self.allocate_buffers(self.engine, grid_sizes) AttributeError: 'TrtYOLO' object has no attribute 'allocate_buffers'

PyCUDA ERROR: The context stack was not empty upon module cleanup.

A context was still active when the context stack was being cleaned up. At this point in our execution, CUDA may already have been deinitialized, so there is no way we can finish cleanly. The program will be aborted now. Use Context.pop() to avoid this problem. `