当前位置:网站首页>pytorch的Dataset的使用
pytorch的Dataset的使用
2022-07-06 09:11:00 【一曲无痕奈何】
from torch.utils.data import Dataset
from PIL import Image
import os
class MyData():
def __init__(self, root_dir, label_dir):
#使用self相当于创建一个全局变量给后面的也可以使用
self.root_dir = root_dir
self.label_dir = label_dir
self.path = os.path.join(self.root_dir, self.label_dir)
self.img_path = os.listdir(self.path)
#获取其中每一个图片
def __getitem__(self, index):
#通过index获取图片地址
img_name = self.img_path[index] #每个图片的路径
img_item_path = os.path.join(self.root_dir, self.label_dir, img_name)
print(img_item_path)
#读取图片
img = Image.open(img_item_path)
img.show
label = self.label_dir
return img, label
def __len__(self): #数据集有多长
return len(self.img_path)
root_dir = "D:\\数据集\\hymenoptera_data\\train"
ant_lable_dair = "ants"
bees_label_dir = "bees"
ant_dataset = MyData(root_dir, ant_lable_dair)
bees_dataset = MyData(root_dir,bees_label_dir)
ant_dataset.__getitem__(0)
for i in range(1,100):
bees_dataset.__getitem__(i)
img, label = ant_dataset[0]
img.show()
边栏推荐
- Sichuan cloud education and double teacher model
- [CV] target detection: derivation of common terms and map evaluation indicators
- What should the redis cluster solution do? What are the plans?
- Some thoughts on the study of 51 single chip microcomputer
- Retention policy of RMAN backup
- CAPL script pair High level operation of INI configuration file
- Implement sending post request with form data parameter
- C杂讲 文件 初讲
- [NLP] bert4vec: a sentence vector generation tool based on pre training
- Use xtrabackup for MySQL database physical backup
猜你喜欢
Embedded development is much more difficult than MCU? Talk about SCM and embedded development and design experience
Mexican SQL manual injection vulnerability test (mongodb database) problem solution
MySQL实战优化高手11 从数据的增删改开始讲起,回顾一下Buffer Pool在数据库里的地位
The appearance is popular. Two JSON visualization tools are recommended for use with swagger. It's really fragrant
13 medical registration system_ [wechat login]
MySQL实战优化高手04 借着更新语句在InnoDB存储引擎中的执行流程,聊聊binlog是什么?
MySQL实战优化高手02 为了执行SQL语句,你知道MySQL用了什么样的架构设计吗?
C miscellaneous shallow copy and deep copy
MySQL storage engine
The underlying logical architecture of MySQL
随机推荐
Preliminary introduction to C miscellaneous lecture document
Redis集群方案应该怎么做?都有哪些方案?
The 32 year old programmer left and was admitted by pinduoduo and foreign enterprises. After drying out his annual salary, he sighed: it's hard to choose
[Julia] exit notes - Serial
MySQL实战优化高手06 生产经验:互联网公司的生产环境数据库是如何进行性能测试的?
Canoe CAPL file operation directory collection
South China Technology stack cnn+bilstm+attention
MySQL實戰優化高手08 生產經驗:在數據庫的壓測過程中,如何360度無死角觀察機器性能?
Target detection -- yolov2 paper intensive reading
112 pages of mathematical knowledge sorting! Machine learning - a review of fundamentals of mathematics pptx
如何让shell脚本变成可执行文件
Tianmu MVC audit II
MySQL combat optimization expert 02 in order to execute SQL statements, do you know what kind of architectural design MySQL uses?
MySQL实战优化高手05 生产经验:真实生产环境下的数据库机器配置如何规划?
15 医疗挂号系统_【预约挂号】
MySQL real battle optimization expert 11 starts with the addition, deletion and modification of data. Review the status of buffer pool in the database
Several errors encountered when installing opencv
Pointer learning
Download address of canoe, download and activation of can demo 16, and appendix of all canoe software versions
Solution to the problem of cross domain inaccessibility of Chrome browser