search

yasenh / libtorch-yolov5

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

Memory leak issues, the program will die #18

Closed molyswu closed 3 years ago

molyswu commented 3 years ago

Hi Memory leak issues, 3 consecutive for ./libtorch-yolov5 --source ../images/bus.jpg --weights ../weights/yolov5s.torchscript.pt --gpu --view-img

If the memory is not released, the program will die.

Thanks

yasenh commented 3 years ago

@molyswu Could elaborate how to replicate the issue? It may related the parser library I added recently