一、Parameter check failed at: ../builder/Network.cpp::addScale::482, condition: shift.count > 0
出现这个问题是因为模型的yolo网络版本与加速过后的yolo网络版本不一致。
比如我的模型是yolov5-5.0,然后我git了yolov5-6.0的tensorrt加速项目,那么就会出现这种报错
解决方案:
删除已经克隆的yolov5-6.0加速项目,重新克隆yolov5-5.0加速项目
二、想要获取的参数数量不对xx expected but xx get
这是因为在执行cmke ..
命令之前,需要修改yololayer.h
文件中的CLASS_NUM
参数,这个参数默认是80(coco数据集中的类别数量),如果是在自己的数据集上训练的模型,需要将CLASS_NUM
参数修改为自己的类别数量,我的数据集只有一个类别,所以将该参数改成1。
【重要!!!!】如果你没有修改
CLASS_NUM
参数就执行了cmake ..
以及make
命令,然后出现了该错误,那么你需要先修改CLASS_NUM
参数,然后把build文件删掉,重新按mkdir build``cd build``cmake ..``make
顺序执行命令,因为cmake生成的文件是依赖yololayer.h
文件的,修改了该文件中的参数就要重新cmake