Error when running the node

fabvio / ld-lsi

Deep learning based lane/freespace detector embedded in ROS node (built for UC3M LSI)

89 stars 18 forks source link

Error when running the node #4

Closed seboz123 closed 4 years ago

seboz123 commented 4 years ago

Hi, I am getting an error running the code. The Error appears in cnn_node.py in line 57: output = self.cnn(input_tensor) The exception thrown is as follows: Cannot obtain output. Exception: size mismatch, m1: [1 x 30720], m2: [7680 x 1024] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:249 Any ideas would be appreciated!

fabvio commented 4 years ago

Hi, what is the input size of your images?

seboz123 commented 4 years ago

The input size is 1280x720

fabvio commented 4 years ago

Try to resize them to 640x360 ;)