search

yasenh / libtorch-yolov5

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

torch::Tensor preds = module.forward({ imgTensor }).toTuple()->elements()[0].toTensor() #27

Closed rrjia closed 3 years ago

rrjia commented 3 years ago

torch::Tensor preds = module.forward({ imgTensor }).toTuple()->elements()[0].toTensor() code error, because torch forward get a GenericList not a Tuple

yasenh commented 3 years ago

@rrjia checkout: https://github.com/yasenh/libtorch-yolov5#faq

rrjia commented 3 years ago

thank you very much ,i found it "Make sure "model.model[-1].export = False" when running export script"

yasenh commented 3 years ago

Yep, give a try and most likely it will solve the issue