yasenh / libtorch-yolov5

A LibTorch inference implementation of the yolov5
MIT License
372 stars 114 forks source link

how to use batch? #61

Open AresamaPi opened 2 years ago

AresamaPi commented 2 years ago

how to inference the 2 images batch? i make tensor_img dim {2,img_inputs.rows,img_inputs.cols,img_input.channels()} and img_imput.data is a vector but it doesn't work