search

yasenh / libtorch-yolov5

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

terminate called after throwing an instance of 'std::bad_alloc' #60

Open sihuo521 opened 2 years ago

sihuo521 commented 2 years ago

error: terminate called after throwing an instance of 'std::bad_alloc' what(): std::bad_alloc infer.sh: line 1: 21934 Aborted (core dumped)

shell: ./libtorch-yolov5 --source path/*.jpg --weights path/best.torchscript --gpu --view-img