数据预处理中Decode使用的疑问

xiao | yang

于 2022-10-26 09:42:53 发布

阅读量164

点赞数

分类专栏：深度学习文章标签： python 开发语言

本文链接：https://blog.csdn.net/xi_xiyu/article/details/127526612

版权

本文探讨在数据预处理过程中遇到的PIL库读取图片与Decode组件的兼容性问题。通过源代码分析，指出在使用PIL读取图片后，可能由于格式转换不当导致错误。解决方案是将读取的PIL图像转换为numpy数组以满足Decode的输入要求。

摘要由CSDN通过智能技术生成

源代码：

import os
from mindspore.dataset import GeneratorDataset
import pandas as pd
from PIL import Image
import mindspore.dataset.vision.py_transforms as py_vision
from mindspore.dataset.transforms.py_transforms import Compose





class EyeLandmarksDataset(GeneratorDataset):

    def __init__(self, excel_file, root_dir, transform=None):
        self.landmarks_frame = pd.read_excel(excel_file)
        self.root_dir = root_dir
        self.transform = transform

    def __len__(self):
        return len(self.landmarks_frame)

    def __getitem__(self, idx):
        img_name = os.path.join(self.root_dir,
                                self.landmarks_frame.iloc[idx,3])
        image = Image.open(img_name)
        if self.transform:
            image = self.transform(image)
        landmarks = self.landmarks_frame.iloc[idx,7:].values   #values函数将表格型数据结构转换成数组
        landmarks = landmarks.astype('float').