YOLOv5-ROS icon indicating copy to clipboard operation
YOLOv5-ROS copied to clipboard

Size Missmatch

Open haniraid opened this issue 1 year ago • 0 comments

raidhani@raidhani-All-Series:~/ros2_ws/src/YOLOv5-ROS/yolov5_ros/yolov5_ros$ ros2 launch yolov5_ros yolov5s_simple.launch.py [INFO] [launch]: All log files can be found below /home/raidhani/.ros/log/2024-09-06-22-11-53-881427-raidhani-All-Series-33297 [INFO] [launch]: Default logging verbosity is set to INFO [INFO] [v4l2_camera_node-1]: process started with pid [33299] [INFO] [yolov5_ros-2]: process started with pid [33301] [v4l2_camera_node-1] [ERROR] [1725675114.305518054] [v4l2_camera]: Failed opening device /dev/video0: No such file or directory (2) [yolov5_ros-2] YOLOv5 🚀 2024-9-2 torch 1.11.0+cpu CPU [yolov5_ros-2] [yolov5_ros-2] Loading config/best.torchscript for TorchScript inference... [yolov5_ros-2] Traceback (most recent call last): [yolov5_ros-2] File "/home/raidhani/ros2_ws/src/YOLOv5-ROS/yolov5_ros/install/yolov5_ros/lib/yolov5_ros/yolov5_ros", line 33, in [yolov5_ros-2] sys.exit(load_entry_point('yolov5-ros', 'console_scripts', 'yolov5_ros')()) [yolov5_ros-2] File "/home/raidhani/ros2_ws/src/YOLOv5-ROS/yolov5_ros/build/yolov5_ros/yolov5_ros/main.py", line 283, in ros_main [yolov5_ros-2] rclpy.spin(yolov5_node) [yolov5_ros-2] File "/opt/ros/foxy/lib/python3.8/site-packages/rclpy/init.py", line 191, in spin [yolov5_ros-2] executor.spin_once() [yolov5_ros-2] File "/opt/ros/foxy/lib/python3.8/site-packages/rclpy/executors.py", line 719, in spin_once [yolov5_ros-2] raise handler.exception() [yolov5_ros-2] File "/opt/ros/foxy/lib/python3.8/site-packages/rclpy/task.py", line 239, in call [yolov5_ros-2] self._handler.send(None) [yolov5_ros-2] File "/opt/ros/foxy/lib/python3.8/site-packages/rclpy/executors.py", line 429, in handler [yolov5_ros-2] await call_coroutine(entity, arg) [yolov5_ros-2] File "/opt/ros/foxy/lib/python3.8/site-packages/rclpy/executors.py", line 354, in _execute_subscription [yolov5_ros-2] await await_or_execute(sub.callback, msg) [yolov5_ros-2] File "/opt/ros/foxy/lib/python3.8/site-packages/rclpy/executors.py", line 118, in await_or_execute [yolov5_ros-2] return callback(*args) [yolov5_ros-2] File "/home/raidhani/ros2_ws/src/YOLOv5-ROS/yolov5_ros/build/yolov5_ros/yolov5_ros/main.py", line 269, in image_callback [yolov5_ros-2] class_list, confidence_list, x_min_list, y_min_list, x_max_list, y_max_list = self.yolov5.image_callback(image_raw) [yolov5_ros-2] File "/home/raidhani/ros2_ws/src/YOLOv5-ROS/yolov5_ros/build/yolov5_ros/yolov5_ros/main.py", line 128, in image_callback [yolov5_ros-2] pred = self.model(im, augment=False, visualize=False) [yolov5_ros-2] File "/home/raidhani/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl [yolov5_ros-2] return forward_call(*input, **kwargs) [yolov5_ros-2] File "/home/raidhani/ros2_ws/src/YOLOv5-ROS/yolov5_ros/build/yolov5_ros/yolov5_ros/models/common.py", line 416, in forward [yolov5_ros-2] y = self.model(im)[0] [yolov5_ros-2] File "/home/raidhani/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl [yolov5_ros-2] return forward_call(*input, **kwargs) [yolov5_ros-2] RuntimeError: The following operation failed in the TorchScript interpreter. [yolov5_ros-2] Traceback of TorchScript, serialized code (most recent call last): [yolov5_ros-2] File "code/torch/models/yolo.py", line 78, in forward [yolov5_ros-2] _36 = (_20).forward(act, _35, ) [yolov5_ros-2] _37 = (_22).forward((_21).forward(act, _36, ), _30, ) [yolov5_ros-2] _38 = (_24).forward(_34, _36, (_23).forward(_37, ), ) [yolov5_ros-2] ~~~~~~~~~~~~ <--- HERE [yolov5_ros-2] return (_38,) [yolov5_ros-2] class Detect(Module): [yolov5_ros-2] File "code/torch/models/yolo.py", line 108, in forward [yolov5_ros-2] _20 = torch.split_with_sizes(torch.sigmoid(_19), [2, 2, 11], 4) [yolov5_ros-2] xy, wh, conf, = _20 [yolov5_ros-2] _21 = torch.add(torch.mul(xy, CONSTANTS.c0), CONSTANTS.c1) [yolov5_ros-2] ~~~~~~~~~ <--- HERE [yolov5_ros-2] xy0 = torch.mul(_21, torch.select(CONSTANTS.c2, 0, 0)) [yolov5_ros-2] _22 = torch.pow(torch.mul(wh, CONSTANTS.c0), 2) [yolov5_ros-2] [yolov5_ros-2] Traceback of TorchScript, original code (most recent call last): [yolov5_ros-2] /home/raidhani/ros2_ws/src/yolov5/models/yolo.py(111): forward [yolov5_ros-2] /home/raidhani/.local/lib/python3.8/site-packages/torch/nn/modules/module.py(1098): _slow_forward [yolov5_ros-2] /home/raidhani/.local/lib/python3.8/site-packages/torch/nn/modules/module.py(1110): _call_impl [yolov5_ros-2] /home/raidhani/ros2_ws/src/yolov5/models/yolo.py(169): _forward_once [yolov5_ros-2] /home/raidhani/ros2_ws/src/yolov5/models/yolo.py(270): forward [yolov5_ros-2] /home/raidhani/.local/lib/python3.8/site-packages/torch/nn/modules/module.py(1098): _slow_forward [yolov5_ros-2] /home/raidhani/.local/lib/python3.8/site-packages/torch/nn/modules/module.py(1110): _call_impl [yolov5_ros-2] /home/raidhani/.local/lib/python3.8/site-packages/torch/jit/_trace.py(958): trace_module [yolov5_ros-2] /home/raidhani/.local/lib/python3.8/site-packages/torch/jit/_trace.py(741): trace [yolov5_ros-2] export.py(274): export_torchscript [yolov5_ros-2] export.py(218): outer_func [yolov5_ros-2] export.py(1414): run [yolov5_ros-2] /home/raidhani/.local/lib/python3.8/site-packages/torch/autograd/grad_mode.py(27): decorate_context [yolov5_ros-2] export.py(1536): main [yolov5_ros-2] export.py(1541): [yolov5_ros-2] RuntimeError: The size of tensor a (80) must match the size of tensor b (60) at non-singleton dimension 3 [yolov5_ros-2] [ERROR] [yolov5_ros-2]: process has died [pid 33301, exit code 1, cmd '/home/raidhani/ros2_ws/src/YOLOv5-ROS/yolov5_ros/install/yolov5_ros/lib/yolov5_ros/yolov5_ros --ros-args --params-file /tmp/launch_params_vrv869j1']. im facing this error notice that i updated the yaml file and also the torchscript at main.py code. and i used this dataset to train the model https://www.kaggle.com/datasets/thepbordin/indoor-object-detection i need solution for this error please

haniraid avatar Sep 07 '24 02:09 haniraid