标签：lable img PyTorch1 self Dataset path root dir 加载

 1 from torch.utils.data import Dataset
 2 from PIL import Image
 3 
 4 import cv2
 5 import os
 6 
 7 class MyData(Dataset):
 8     def __init__(self, root_dir, lable_dir):
 9         self.root_dir = root_dir
10         self.lable_dir = lable_dir
11         self.path = os.path.join(self.root_dir, self.lable_dir)
12         self.img_path = os.listdir(self.path)
13 
14     def __getitem__(self, idx):
15         img_name = self.img_path[idx]
16         img_item_path = os.path.join(self.root_dir, self.lable_dir, img_name)
17         img = Image.open(img_item_path)
18         lable = self.lable_dir
19         return img, lable
20 
21     def __len__(self):
22         return len(self.img_path)
23 
24 root_dir = "E:\\python\\learn_pytorch\\dataset\\train"
25 ants_lable_dir = "ants"
26 bees_lable_dir = "bees"
27 ants_dataset = MyData(root_dir, ants_lable_dir)
28 bees_dataset = MyData(root_dir, bees_lable_dir)

标签：lable,img,PyTorch1,self,Dataset,path,root,dir,加载
来源： https://www.cnblogs.com/0xiaoyu/p/16119206.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。