首页 PaddleCV 帖子详情
图像识别生成reader时出错
收藏
快速回复
PaddleCV 问答目标识别 958 1
图像识别生成reader时出错
收藏
快速回复
PaddleCV 问答目标识别 958 1
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
收藏
回复
全部评论(1)
时间顺序
AIStudio810261
#2 回复于2020-04

这个是在AIStudio上吗? 如果在本地一切正常, 到了AIStudio出现换行, 估计是前端渲染没画对吧. 但这个没啥影响的. 核心是你训练出来的模型. 

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户