深度学习tensorflow框架在训练的时候,sess.run(feed_dict={image: x_tr, label: y_tr, lr: learning_rate})遇到的报错。
我所使用到的解决方案是:
1、图像size大小不匹配(我花了2个小时找,终于解决)
2、图像数量不匹配(我写了一个程序判断图像数量是否对应和包含)
# coding=utf-8
import os
# 将文件A和B中不同的文件找出来
path1=os.getcwd()
images_path='dataset/little_sample/val/labels_binary'
labels_path='dataset/little_sample/val/images'
Image_pathDir = os.listdir(images_path)
len_images=len(Image_pathDir)
labels_pathDir = os.listdir(labels_path)
len_labels=len(labels_pathDir)
image_P = []
label_P = []
num=0
for i in range(len(Image_pathDir)):
image_name=Image_pathDir[i]
if image_name not in labels_pathDir:
num=num+1
print(image_name)
print(num)
if(num==0):
print('Labels is same to images ')
else:
print("Image not match!")
num1=0
for i in range(len(labels_pathDir)):
image_name=labels_pathDir[i]
if image_name not in Image_pathDir:
num1=num1+1
print(image_name)
print(num1)
if num1==0:
print('Images is same to labels')
else:
print("Image not match!")