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)
@f3rm4rf3r updated