找回密码
 立即注册
发表于 2021-4-4 02:11:56
4434
在用yolo开始训练时报错:连接”层要求输入的形状与连接轴一致,但形状必须匹配,百度后还是不知道如何操作


1.png
2.png
使用道具 举报 回复
发表于 2021-4-4 14:53:17
我刚才又试了几遍tlt3.0 yolo的部分,没发现您的错误,能麻烦你提供一下,您训练用到的yolo_train_resnet18_kitti.txt吗?
使用道具 举报 回复 支持 反对
发表于 2021-4-4 15:05:36
本帖最后由 myf636466 于 2021-4-4 15:07 编辑
nvadmin 发表于 2021-4-4 14:53
我刚才又试了几遍tlt3.0 yolo的部分,没发现您的错误,能麻烦你提供一下,您训练用到的yolo_train_resnet18 ...

在出错之后尝试过修改big_anchor_shape,mid_anchor_shape,small_anchor_shape。是按照下边图片所要求修改的

3.png
4.png
5.png
使用道具 举报 回复 支持 反对
发表于 2021-4-4 15:27:55
数据集是我自己准备的,会不会yolo输入的问题?我用ssd去训练都没问题
使用道具 举报 回复 支持 反对
发表于 2021-4-4 23:13:29
以解决,是数据集的问题,yolo的输入必须是32的倍数。
6.png
使用道具 举报 回复 支持 反对
发新帖
您需要登录后才可以回帖 登录 | 立即注册