search

yasenh / libtorch-yolov5

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

No obj problem #11

Closed name333 closed 4 years ago

name333 commented 4 years ago

hello , Let me post a question in your project. In detect.cpp, // if none remain then process next image if (det.size(1) == 0) { continue; } det.size(1) == 0　should be　det.size(0) == 0　Is that right？

yasenh commented 4 years ago

Hi @name333, thanks for pointing that out and it was fixed in my previous commits: cc7887c You could pull from the latest master.