__getitem__的理解以及实现

一、__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)