图像识别生成reader时出错
收藏
def train_mapper(sample): #print(sample) img_path,label=sample try: img=Image.open(img_path) #进行了简单的图像变换,这里对图像进行crop修剪操作,输出img的维度为(3, 100, 100) img = paddle.dataset.image.simple_transform(im=img, #输入图片是HWC resize_size=100, # 剪裁图片 crop_size=100, is_color=True, #彩色图像 is_train=True) #将img数组进行进行归一化处理,得到0到1之间的数值 img= img.flatten().astype('float32')/255.0 return img,int(label) except: pass def train_r(train_list,buffered_size=1024): def reader(): with open(train_list,'r') as f: lines=f.readlines() for line in lines: img_path,label=line.split('\t') yield img_path,label return paddle.reader.xmap_readers(train_mapper,reader,cpu_count(),buffered_size) trainer_reader = train_r(train_list="data/data10954/train.txt") temp_reader = paddle.batch(trainer_reader, batch_size=4) temp_data=next(temp_reader()) print(temp_data)
yield img_path,label 之前 print(img_path,label)结果是
但是在传入train_mapper后打印sample结果是:
会出现这种有时未换行的情况。为什么会出现这种结果,是我哪里写的有问题吗?
0
收藏
请登录后评论
这个是在AIStudio上吗? 如果在本地一切正常, 到了AIStudio出现换行, 估计是前端渲染没画对吧. 但这个没啥影响的. 核心是你训练出来的模型.