a bug when use in libtorch 1.7.1 version

wsx000 / YOLOv4-LibTorch-GPU-deployment

This is a YOLOv4 post-processing code used gpu based on pure libtorch API

GNU General Public License v2.0

14 stars 7 forks source link

Closed happyboyneu closed 3 years ago

happyboyneu commented 3 years ago

line 125 in yolo.cpp,there should be a parentheses before .unsqueeze(-1). like this

wsx000 commented 3 years ago

thanks for your discovery! I have corrected it