有没有大佬帮忙看看我这样自定义数据集为什么有问题
藐视亦切 发布于2020-06-03 11:24 浏览:81 回复:2
0
收藏

image_path = "E:/kaggle/data/*" #文件里已经标注好的图片
test_path = "E:/kaggle/data/*"

def load_image(file):
im = Image.open(file) #打开图像
im = im.resize((224,224),Image.ANTIALIAS) #把图像大小改为224*224
im = np.array(im).reshape([3,224,224]).astype(np.float32) #返回新形状的数组,把它变成一个 numpy 数组以匹配数据馈送格式。
im = im / 255.0 * 2.0 - 1.0 #归一化
return im
train_list = []
test_list = []
for idx in glob.iglob(image_path):
train_list.append(idx)
for idx in glob.iglob(test_path):
test_list.append(idx)
print(train_list)
def train_reader_createor(path): #读取训练数据
def __reader__():
for idx in range(len(train_list)):
image = load_image(train_list[idx])
yield image, train_lables[idx]
return __reader__
train_reader = paddle.batch(
paddle.reader.shuffle(reader=train_reader_createor(image_path),
buf_size=BUF_SIZE),
batch_size=BATCH_SIZE)

收藏
点赞
0
个赞
共2条回复 最后由星光ld1回复于2020-06-03 23:41
#3星光ld1回复于2020-06-03 23:41:30

先不说有没有其他问题就这段代码

def train_reader_createor(path): #读取训练数据
  def __reader__():
    for idx in range(len(train_list)):
    image = load_image(train_list[idx])
    yield image, train_lables[idx]
  return __reader__

path意义何在。。。直接使用了上面的全局train_list

0
#2鹿鼎记肯定回复于2020-06-03 22:18:49

哪里报错了?你的代码可以用代码编辑器打印出来

0
TOP
切换版块