一、__getitem__的理解以及实现

__getitem__的作用是用来给予[]实现的,有两种方式:

  1. 通过[]去访问;
  2. 对对象进行访问,作为一个迭代对象访问,可以回想一下,数组就是这么访问的,同时数组也是通过[]访问,所以实际可以把对象看做一个数组;

二、示例

示例1

一个简单的示例如下:

# __getitem__怎么使用
class Animal:
    def __init__(self, animal_list,age):
        self.animals_name = animal_list
        self.animals_age=age
        
    def __getitem__(self, index):
        return self.animals_name[index]
 
animals = Animal(["dog","cat","fish"],[1,2,3])
for i in animals:
    print(i)

在这里插入图片描述

示例2

class 类:
{
    def __getitem__(self, index): #进行迭代调用就是用的这个函数
        # 根据索引index从文件中读取一个数据
        # 对数据预处理
        # 返回数据和对应标签
        img_path, label = self.imgs_info[index]
        img = Image.open(img_path) #读取出来的是值是RGBA,其中A暂时是没有用的,所以需要进行转换
        img = img.convert('RGB')  #将其转换为RGB的格式
        img = self.padding_black(img) #将图片进行填充,使其大小为224*224
        if self.train_flag:
            img = self.train_tf(img)
        else:
            img = self.val_tf(img)
        label = int(label)

        return img, label
}

for image, label in train_loader:
        print(image.shape)
        print(image)
        # img = transform_BZ(image)
        # print(img)
        print(label)
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐