在用yolo开始训练时报错:连接”层要求输入的形状与连接轴一致,但形状必须匹配,百度后还是不知道如何操作
我刚才又试了几遍tlt3.0 yolo的部分,没发现您的错误,能麻烦你提供一下,您训练用到的yolo_train_resnet18_kitti.txt吗?
在出错之后尝试过修改big_anchor_shape,mid_anchor_shape,small_anchor_shape。是按照下边图片所要求修改的
数据集是我自己准备的,会不会yolo输入的问题?我用ssd去训练都没问题
以解决,是数据集的问题,yolo的输入必须是32的倍数。