DeepLearing—CV系列(一)——猫狗识别的Pytorch实现
文章目录一、理论二、代码2.1 Mydataset.py2.2 Mynet.py2.3 Train.py一、理论二、代码2.1 Mydataset.pyfrom torch.utils.data import DataLoader,Datasetimport osfrom PIL import Imageimport torchimport numpy as npfrom torchvision.transforms import ToPILImage,ToTensorclass My
复制链接