yolov3/yolov5 val.py训练正常测试错误

基于u版的yolov3/yolov5上搭建自己的模型，修改主干网络或者head，训练可以正常训练，训练完成后用命令python val.py --weights runs/exp/train/best.pt进行测试，报错：
RuntimeError: Sizes of tensors must match except in dimension 2. Got 27 and 28 (The offending index is 0)
如下图所示

开始以为是维度错误或者特征图大小不匹配，但能正常训练，应该不是搭建的网络有问题，最后在github/yolov5找到了解决方法：
定位common.py，将#stride = int(model.stride.max())修改为stride = max(int(model.stride.max()), 32)，如下：

elif pt:  # PyTorchfrom models.experimental import attempt_load  # scoped to avoid circular importmodel = torch.jit.load(w) if 'torchscript' in w else attempt_load(weights, map_location=device)#stride = int(model.stride.max())  # model stridestride = max(int(model.stride.max()), 32)  # model stride

yolov3/yolov5 val.py训练正常测试错误相关推荐

yolov5 v3.0训练出现KeyError错误
欢迎大家关注笔者,你的关注是我持续更博的最大动力原创文章,转载告知,盗版必究 yolov5 v3.0训练出现KeyError错误文章目录: 1 错误原因分析 2 错误解决方式 1 错误原因分析出 ...
Yolov5自定义图片训练测试及模型调优（详细过程）
Yolov5自定义图片训练测试及模型调优(详细过程) 1. 图像数据获取.标注 2. 数据集划分及准备 3. 配置训练参数,准备训练: 3.1 修改data/训练配置xxx.yaml文件: 3.2 修 ...
基于人脸的常见表情识别——模型搭建、训练与测试¶
整个训练流程包括数据接口准备.模型定义.结果保存与分析. 数据接口一般使用torchvision.Dataset定义数据的读取.torch.utils.data.Dataloader定义数据的加载. ...
天池训练营——基于人脸的常见表情识别(3)——模型搭建、训练与测试
在完成数据准备之后,便可以使用 PyTorch 深度学习框架,实现卷积神经网络的定义.训练和预测. 一.模型搭建与训练得到了数据之后,接下来咱们使用 PyTorch 这个框架来进行模型的训练.整个训 ...
基于人脸的常见表情识别（3）——模型搭建、训练与测试
基于人脸的常见表情识别(3)--模型搭建.训练与测试模型搭建与训练 1. 数据接口准备 2. 模型定义 3. 模型训练模型测试本 Task 是『基于人脸的常见表情识别』训练营的第 3 课,如果你 ...
YOLOv5的Tricks | 【Trick14】YOLOv5的val.py脚本的解析
如有问题,恳请指出. 这篇可能是这个系列最后的一篇了,最后把yolov5的验证过程大致的再介绍介绍,基本上把yolov5的全部内容就稍微过了一遍了,也是我自己对这个项目学习的结束.(补充一下,这里我介 ...
自定义ava数据集及训练与测试完整版时空动作/行为视频数据集制作 yolov5, deep sort, VIA MMAction, SlowFast
前言这一篇博客应该是我花时间最多的一次了,从2022年1月底至2022年4月底. 我已经将这篇博客的内容写为论文,上传至arxiv:https://arxiv.org/pdf/2204.10160. ...
yolov5 test.py val.py detec.py 区别在哪里呢?
yolov5 test.py val.py detec.py 区别在哪里呢? 用户在训练数据的时候必须使用 train.py 来进行数据训练和验证,但我很难理解 detect.py 和 test.p ...
Windows下使用Yolov3（GPU）训练+测试自己的数据集
Windows下使用Yolov3(GPU)训练+测试自己的数据集 1.配置Yolov3 参考:Windows下使用darknet.exe跑通Yolov3 Window10+VS2017+CUDA10. ...

yolov3/yolov5 val.py训练正常测试错误

yolov3/yolov5 val.py训练正常测试错误相关推荐

最新文章

热门文章