读取数据后预处理报错

最新推荐文章于 2023-05-05 09:03:08 发布

xiao | yang

最新推荐文章于 2023-05-05 09:03:08 发布

阅读量194

点赞数

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

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

版权

在实现数据读取和预处理的过程中，数据读取部分运作正常，但在执行预处理步骤时遇到错误。报错提示不推荐继承使用GeneratorDataset进行构造。这可能涉及到对GeneratorDataset的不当使用或理解不足。

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

import os
from mindspore.dataset import GeneratorDataset
import pandas as pd
from mindspore import Tensor
from mindspore import dtype as mstype
import mindspore.dataset.vision.c_transforms as CV
import mindspore.dataset.transforms.c_transforms as C
from mindspore.dataset.vision import Inter
import cv2

"""数据读取类和预处理函数定义"""
"""数据读取，返回的是一个字典，字典内数据类型为张量"""
class EyeLoad(GeneratorDataset):

    def __init__(self, excel_file, root_dir):
        #super(EyeLoad, self).__init__()
        self.landmarks_frame = pd.read_excel(excel_file, engine='openpyxl')
        self.root_dir = root_dir

    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])
       # with cv2.imread(img_name, "rb") as cr:
        image = cv2.imread(img_name)
        la