python自己写一个类_python 写一个自己的类入门

1.类和函数不同，对于一个类，在初始化过程中，我们需要给初始变量，进行初始化，这类似于C++的构造函数，初始化的用法一般是init(self, a,b,c)：这里a,b,c是我们呢自己输入的初始变量，

#coding:utf-8

from PIL import Image

from torch.utils.data import Dataset

class MyDataset(Dataset):

def __init__(self,txt_path,transform=None,target_transform=None):

fh=open(txt_path,'r')

imgs=[]

for line in fh:

line=line.rstrip()

words=line.split()

imgs.append((words[0],int(words[1])))

self.imgs=imgs

self.transform=transform

self.target_transform=target_transform

def __getitem__(self,index):

fn,label=self.imgs[index]

img =Image.open(fn).convert('RGB')

if self.transform is not None:

img=self.transform(img)#在这里做图像的转换，转换到(0，1)

return img,label

def __len__(self):

return len(self.imgs)

这里第一步都类,继承了基类，这里暂时不做展开.

在这里后面，self.img,类似的是添加内部变量，这里类似C++的变量分类。每一个函数都有对应的操作。