yasenh / libtorch-yolov5

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

When `PostProcessing()` error occurred #49

Open blackCmd opened 3 years ago

blackCmd commented 3 years ago

image

First step, PostProcess() with temp_img is fine. But second step, PostProcess() with my custom image the error occurred

ㄴ When I set half_ to false, the error has gone. Why this error occured? How to run this code with half_