tf报错 之 assertion failed: [Need value.shape >= size, got ] [336 210 3] [224 224 3]

根据信息提示,我们定位到了报错函数,函数原型:tf.image.random_crop(image,size,seed=None,name=None)

产生这个问题的原因是:图片原来的 height 或 width 要小于 你设置的裁剪值 的大小

解决办法:
1.将裁剪值调小,但如果你的数据集比较大的话,这种方法可能要用很多次,才能找到最合适的最小值
2.先统一将图片resize成一个统一的大小,在进行裁剪,这样就方便不少

resize函数:tf.image.resize(image,size)

猜你喜欢

转载自blog.csdn.net/qq_41368074/article/details/110693600