tensorrtx: ./yolov5s -s Fails

Cannot load weights I have yolov5s.wts in the root folder and building from the build folder. This is weird and should not happen.

root@0da1a0f1cdeb:/home/build# ls ..
CMakeLists.txt  build       gen_wts.py  utils.h       yololayer.h  yolov5s.wts
README.md       common.hpp  logging.h   yololayer.cu  yolov5s.cpp

root@0da1a0f1cdeb:/home/build# ./yolov5s -s
Loading weights: ../yolov5s.wts
[06/23/2020-06:50:46] [E] [TRT] Parameter check failed at: ../builder/Network.cpp::addScale::434, condition: shift.count > 
0 ? (shift.values != nullptr) : (shift.values == nullptr)
yolov5s: /home/common.hpp:189: nvinfer1::IScaleLayer* addBatchNorm2d(nvinfer1::INetworkDefinition*, 
std::map<std::__cxx11::basic_string<char>, nvinfer1::Weights>&, nvinfer1::ITensor&, std::__cxx11::string, float): Assertion 
`scale_1' failed.
Aborted (core dumped)

About this issue

  • Original URL
  • State: closed
  • Created 4 years ago
  • Comments: 32 (25 by maintainers)

Most upvoted comments