Description
使用jetson nano Ubuntu18.04 部署yolov5时，生成yolov5s.engine文件报错
Environment
Ubuntu Version: 18.04
Python Version (if applicable): 3.6
PyTorch Version (if applicable): 1.8
YoLov5 Version: V5
Relevant Files
sudo ./yolov5 -s yolov5s.wts yolov5s.engine s
Loading weights: yolov5s.wts
[11/10/2021-20:28:57] [E] [TRT] Parameter check failed at: …/builder/Network.cpp::addScale::482, condition: shift.count > 0 ? (shift.values != nullptr) : (shift.values == nullptr)
yolov5: /home/nano/tensorrtx/yolov5/common.hpp:155: nvinfer1::IScaleLayer* addBatchNorm2d(nvinfer1::INetworkDefinition*, std::map<std::__cxx11::basic_string, nvinfer1::Weights>&, nvinfer1::ITensor&, std::__cxx11::string, float): Assertion `scale_1’ failed.
Aborted
NVES
2
Hi,
This looks like a Jetson issue. Please refer to the below samlples in case useful.
For any further assistance, we recommend you to raise it to the respective platform from the below link
Thanks!
father
5
有解决吗我转换的时候也是这个错误sudo ./yolov5 -s yolov5l.wts yolov5l.engine l
Loading weights: yolov5l.wts
[12/27/2022-16:12:05] [E] [TRT] Parameter check failed at: …/builder/Network.cpp::addScale::482, condition: shift.count > 0 ? (shift.values != nullptr) : (shift.values == nullptr)
yolov5: /home/nvidia/yolov5/tensorrtx/yolov5/common.hpp:155: nvinfer1::IScaleLayer* addBatchNorm2d(nvinfer1::INetworkDefinition*, std::map<std::__cxx11::basic_string, nvinfer1::Weights>&, nvinfer1::ITensor&, std::__cxx11::string, float): Assertion `scale_1’ failed.
father
6
system
Closed
8
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.