__getitem__的理解以及实现
【代码】__getitem__的理解以及实现。
·
一、__getitem__的理解以及实现
__getitem__的作用是用来给予[]实现的,有两种方式:
- 通过[]去访问;
- 对对象进行访问,作为一个迭代对象访问,可以回想一下,数组就是这么访问的,同时数组也是通过[]访问,所以实际可以把对象看做一个数组;
二、示例
示例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)
更多推荐
已为社区贡献1条内容
所有评论(0)