search

yasenh / libtorch-yolov5

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

terminated call error when run /libtorch-yolov5 #35

Closed alikarimi120 closed 3 years ago

alikarimi120 commented 3 years ago

Capture

alikarimi120 commented 3 years ago

i sure "model.model[-1].export = False" in export.py

alikarimi120 commented 3 years ago

Can you help? @yasenh Which yolov5 release did you use? https://github.com/ultralytics/yolov5/releases

yasenh commented 3 years ago

I tested it on v3.0 and v3.1 @alikarimi120 , are you using v4.0? And which version of libtorch are you using?

alikarimi120 commented 3 years ago

I tested it on v3.0 and v3.1 @alikarimi120 , are you using v4.0? And which version of libtorch are you using?

libortch : https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.6.0.zip

i tested it on v2.0 , v3.1 and v4 but I get this error (picture above)

alikarimi120 commented 3 years ago

Can you send me a yolov5s.torchscript.pt? of course, I think the problem is with the pytorch version, not the model conversion? @yasenh

alikarimi120 commented 3 years ago

I've tried nigthly version of libtorch and error dissapear command: wget https://download.pytorch.org/libtorch/nightly/cu101/libtorch-cxx11-abi-shared-with-deps-latest.zip

look it : https://github.com/pytorch/pytorch/issues/17054

yasenh commented 3 years ago

libtorch >= 1.6.0 should work