search

DanaHan / Yolov5-in-Deepstream-5.0

Describe how to use yolov5 in Deepstream 5.0
209 stars 57 forks source link

issues with serializing into engine file #29

Open lchunleo opened 2 years ago

lchunleo commented 2 years ago

Hi

i having issue trying to convert a generated wts (based on 3.1) into engine file when executing yolov5. Can anyone advise?

oading weights: yolov5l_3.1.wts [10/13/2021-14:41:31] [E] [TRT] Parameter check failed at: ../builder/Network.cpp::addScale::494, condition: shift.count > 0 ? (shift.values != nullptr) : (shift.values == nullptr) yolov5: /home/yolov5/common.hpp:155: nvinfer1::IScaleLayer addBatchNorm2d(nvinfer1::INetworkDefinition, std::map<std::__cxx11::basic_string, nvinfer1::Weights>&, nvinfer1::ITensor&, std::string, float): Assertion `scale_1' failed. Aborted (core dumped)

Emirismail commented 1 year ago

@lchunleo have you managed to solve this please?