跟着小土堆学习pytorch(一)——Dataset

文章目录

  • 一、前言
  • 二、dataset
  • 三、代码展示

一、前言

pytorch也是鸽了很久了,确定了下,还是用小土堆的教程。

kaggle获取数据集
在这里插入图片描述

二、dataset

dateset:数据集——提供一种方式去获取数据及其标签
如何获取数据及其标签以及总共多少个数据
dataloder:数据装载器——为后面的网络提供不同的数据形式
关于dataset也是重写类

三、代码展示

from torch.utils.data import Dataset
from PIL import Image
import osclass MyData(Dataset):def __init__(self, root_dir, label_dir):self.root_dir = root_dirself.label_dir = label_dirself.path = os.path.join(self.root_dir,self.label_dir)self.img_path = os.listdir(self.path)'''root_dir 根目录label_dir 路径os.path.join是为了把这两个路径加起来,用了这个函数后就不会出错了os.listdir这个是吧路径下的一系列文件名变成列表'''def __getitem__(self, idx):img_name = self.img_path[idx]img_item_path = os.path.join(self.root_dir,self.label_dir,img_name)img = Image.open(img_item_path)label = self.label_dirreturn img,label# 给getitem一个索引,就会返回img和labeldef __len__(self):return len(self.img_path)root_dir = "dataset/train"
ants_label_dir = "ants"
bees_label_dir = "bees"
bees_dataset = MyData(root_dir,bees_label_dir)
ants_dataset = MyData(root_dir,ants_label_dir)train_dataset = ants_dataset + bees_dataset
# 两个数据集的集合# img_path = "D:\\pycharm_project\\try3\\dataset\\train\\ants\\0013035.jpg"
#
# image = cv2.imread(img_path)
#
# if image is not None:
#     # 如果成功读取,显示图片
#     cv2.imshow('Image', image)
#     cv2.waitKey(0)  # 等待按键
#     cv2.destroyAllWindows()  # 关闭所有窗口
# else:
#     print("Error: 图片读取失败。请检查图片路径。")

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/bicheng/56390.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

PostgreSQL学习笔记六:模式SCHEMA

模式(Schema) PostgreSQL中的模式(Schema)是一个命名的数据库对象集合,包括表、视图、索引、数据类型、函数、存储过程和操作符等。模式的主要作用是组织和命名空间数据库对象,使得同一个数据库中可以包含…

基于gewechat制作第一个微信聊天机器人

Gewe 个微框架 GeWe(个微框架)是一个创新性的软件开发框架,为个人微信号以及企业信息安全提供了强大的功能和保障。GeWe的设计旨在简化开发过程,使开发者能够高效、灵活地构建和定制通信协议,以满足不同应用场景的需求…

JavaScript object(2)

这样的话,就变成只读了。

外包干了5天,技术明显退步

我是一名本科生,自2019年起,我便在南京某软件公司担任功能测试的工作。这份工作虽然稳定,但日复一日的重复性工作让我逐渐陷入了舒适区,失去了前进的动力。两年的时光匆匆流逝,我却在原地踏步,技术没有丝毫…

“八股文”面试:助力、阻力还是空谈?

在当今的IT行业,面试程序员时提及“八股文”已成为一种普遍现象。所谓“八股文”,通常指的是一系列固定的、标准化的面试问题及其解答,这些问题往往涵盖了计算机科学和软件工程的基础知识,以及一些流行的技术框架和算法。然而&…

Qualitor checkAcesso.php 任意文件上传漏洞复现(CVE-2024-44849)

0x01 漏洞概述 Qualitor 8.24及之前版本存在任意文件上传漏洞,未经身份验证远程攻击者可利用该漏洞代码执行,写入WebShell,进一步控制服务器权限。 0x02 复现环境 FOFA:app="Qualitor-Web" 0x03 漏洞复现 PoC POST /html/ad/adfilestorage/request/checkAcess…

【IC验证】随机约束

1.约束 (1)注意 一般随机约束只能在类中使用; (2)实现步骤 在定义变量时,用rand/randc关键字进行修饰; 定义约束; 创建并实例化类后,调用随机约束方法; &am…

光平面标定代码

本篇文章主要给出光平面标定代码,鉴于自身水平所限,如有错误,欢迎批评指正。(欢迎进Q群交流:874653199) 数据分为棋盘格数据和激光条数据,激光条数据为在第22个位姿至第26个位姿下打在棋盘格标定…

初识Linux之指令(二)

一:head指令 head 与 tail 就像它的名字一样的浅显易懂,它是用来显示开头或结尾某个数量的文字区块,head 用来显示档案的 开头至标准输出中,而 tail 想当然尔就是看档案的结尾。 语法:head 【参数】 【文件】 功能&…

java Execl 写入png图片 解决

1、先拿到Execl 的所有图片 放到list /*** 获取图片和位置 (xls)** param sheet* return* throws IOException*/public Map<String, PictureData> getPictures(XSSFSheet sheet) throws IOException {Map<String, PictureData> map new HashMap<String, Pict…

開發 meshtastic 聊天機器人(2)

利用 Web 串接主機附近周邊藍芽(含 meshtastic client) pip install bleak (這個比較簡單) ----另外一個為 pybluez2 (pybluez) 2.程式 import streamlit as st import asyncio from bleak import BleakScannerasync def fetch_data():devices await BleakScanner.discover(…

尚硅谷rabbitmq2024 第15-18节 springboot整合与可靠性答疑

在spring boot项目中&#xff0c;只引入了一个amqp的starter&#xff0c;为什么在写listener的时候能看到rabbitmq相关的类&#xff0c;比如RabbitListener( public void processMessage(String dataString, Message message, channel channel){ 这里的Message就是rabbitmq下面…

grpc和http的区别

当面试官问到gRPC和HTTP的区别时&#xff0c;你可以从以下几个方面进行回答&#xff1a; 协议层次&#xff1a; gRPC&#xff1a;gRPC是一个现代的、开源的、高性能的远程过程调用&#xff08;RPC&#xff09;框架&#xff0c;它使用HTTP/2作为传输协议&#xff0c;并使用Pro…

【ICPC】The 2021 CCPC Weihai Onsite G

Shinyruo and KFC #组合数学 #暴力 #枚举 题目描述 During your participation in this competition, Shinyruo is preparing to order KFC for the offline competition next week. There are n n n kinds of foods in KFC, and he plans to order a i a_i ai​ number …

京存非编存储走进中央民族大学

中央民族大学是一所具有鲜明民族特色的综合性全国重点大学&#xff0c;是国家民委、教育部、北京市共建高校。学校前身为1941年成立的延安民族学院。新中国成立后&#xff0c;经中央政府批准&#xff0c;1951年在北京成立中央民族学院&#xff0c;1993年11月更名为中央民族大学…

LeetCode 3158.求出出现两次数字的 XOR 值:化整数为哈希,一次遍历O(1)空间解决

【LetMeFly】3158.求出出现两次数字的 XOR 值&#xff1a;化整数为哈希&#xff0c;一次遍历O(1)空间解决 力扣题目链接&#xff1a;https://leetcode.cn/problems/find-the-xor-of-numbers-which-appear-twice/ 给你一个数组 nums &#xff0c;数组中的数字 要么 出现一次&a…

golang中如何读取nacos配置中心的信息

golang中如何读取nacos配置中心的信息 golang中如何读取nacos配置中心的信息读取步骤1. 安装 Nacos Go 客户端&#xff1a;2. 编写代码读取配置&#xff1a; 代码说明运行代码注意事项 golang中如何读取nacos配置中心的信息 在 Go 语言中读取 Nacos 配置信息&#xff0c;以下是…

【量化交易】聚宽安装

安装JQData 更换源&#xff1a; 如果使用的是pip默认的PyPI源&#xff0c;可以尝试更换为一个更快的国内镜像源。例如阿里云、豆瓣等提供的PyPI镜像。 更改方法可以通过设置环境变量或者在pip命令中直接指定&#xff1a; PS C:\Users\bilirjs\Documents> pip config set …

No.13 笔记 | 网络安全防护指南:从法律法规到技术防御

一、法律法规 《中华人民共和国网络安全法》要点 遵守法律&#xff1a;所有个人和组织在使用网络时&#xff0c;必须遵守宪法和法律&#xff0c;不得利用网络从事危害国家安全等活动。 个人信息保护&#xff1a;禁止非法获取、出售或提供个人信息。若违反但未构成犯罪&#x…

Oracle低代码平台apex介绍

Oracle APEX&#xff08;Application Express&#xff09;是一个强大的低代码开发平台&#xff0c;它允许开发者快速构建企业级Web应用程序。该平台基于Oracle数据库&#xff0c;并充分利用了数据库的功能来提供安全、可扩展且易于维护的应用程序。 什么是Oracle APEX&#xf…