pytorch / android-demo-app

PyTorch android examples of usage in applications
1.46k stars 604 forks source link

Android App crash after click DETECT button #225

Open blackCmd opened 2 years ago

blackCmd commented 2 years ago

I downloaded yolov5s.

And I exported yolov5s.torchscript.ptl using python export.py --weights yolov5s.pt --img 320 --include torchscript.

Then I got error when I click 'detect' button like below. image

The error line is here. image

ps. I changed mOutputRow value 25200to6300in PrePostProcessor.java because I exported 320 not 640