卷积神经网络--猫狗系列之下载、导入数据集

(由于是学习，所以文章会有一些报错及解决办法)

在Kaggle()获取数据集：（没有账号先去注册一个账号，在注册时可能会出现的问题见Kaggle注册出现一排“Captcha must be filled out.”！）

https://www.kaggle.com/competitions/dogs-vs-cats/data

获取到的数据集：(后面几期再分享)

一共25000张图片，猫猫狗狗各占一半。图片名称为："cat.x.jpg"

对图片名称进行处理：

import ospath="D:\\Users\\Twilight\\PycharmProjects\\catdog\\train\\"filenames=[name for name in os.listdir(path)]j=0k=0catList=[]dogList=[]for i,filename in enumerate(filenames):    src=path+filename    namelist=filename.split('.')    if namelist[0]=='cat':        j=j+1        dst=namelist[0]+str(j)+'.0'+'.'+namelist[2] #猫标签设置为0        catList.append(dst)#获得cat的图片名集合    else:        k=k+1        dst=namelist[0]+str(k)+'.1'+'.'+namelist[2] #狗标签设置为1        dogList.append(dst)#获得dog的图片名集合    dst=path+dst    os.rename(src,dst)

这里报错了，解决一下：

emm，就是代码第二行两个反斜杠忘加了：

path="D:\\Users\\Twilight\\PycharmProjects\\Resnet18\\train\\"

运行完后：

这里改名称只是习惯问题，重要的是标签问题，0代表猫猫，1代表狗狗，指明图片的类别，并在构建Dataset类时发挥作用。另外，catList和dogList分别存储了猫猫和狗狗的图片名称的路径，是为了划分训练集和测试集。

紧接着，实现自己的Dataset类：

代码：（训练集和测试集按8：2）

import torchimport osfrom torch.utils.data import Datasetfrom torchvision import transformsfrom PIL import Imageimport numpy as npclass MyDataset(Dataset):    def __init__(self,path_file,namelists,transform=None):        self.path_file=path_file        self.imgs=namelists        self.transform=transform    def __len__(self):        return len(self.imgs)    def __getitem__(self, idx):        #get the image        img_path = os.path.join(self.path_file,self.imgs[idx])        image=Image.open(img_path)        image=image.resize((28,28))#修改图片大小，默认大小        if self.transform:            image = self.transform(image)        #get the label        str1=self.imgs[idx].split('.')        label=eval(str1[1])        return image, label

这个报错之前有个类似的，在anoconda下载，前面有文章写到：PyTorch安装

然后导入DataLoader:

train_loader=torch.utils.data.DataLoader(train_data, batch_size=32, shuffle=True) test_loader=torch.utils.data.DataLoader(test_data, batch_size=32, shuffle=True)

运行完了测试一下：

很好，没有问题。（如果有问题可能就是前面多次调试、运行代码造成的，可以重新创一个项目来运行。）【如果有时间再讲解代码】

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处： http://www.mfbz.cn/a/34743.html 如若内容造成侵权/违法违规/事实不符，请联系我们进行投诉反馈qq邮箱[email protected]，一经查实，立即删除！随着科技的不断进步，人工智能已经成为各个领域中的重要驱动力。在航天领域中，人工智能的应用正日益展现出巨大的潜力。航天领域对精确性、自动化和高效性的需求，使得人工智能成为实现这些目标的关键技术之一。人工智能正在以其独特的优势和算… 阅读更多... 说明：准备Linux系统centos7版本(以上) 2.切换管理模式说明：输入一下命令，然后回车，输入密码。 su – 3.更新yum 说明：为了保证doker能够给顺利安装，那么更新一下；如果没有也可以… 阅读更多... （1）基本介绍 AIPRM for ChatGPT是一个Chrome浏览器扩展程序，基于Chromium内核开发的浏览器都可以使用该扩展，比如微软的Edge浏览器等。在AIPRM的帮助下，我们可以在ChatGPT中一键使用各种专门为网站SEO、SaaS、营销、… 阅读更多... 转载请注明出处：https://blog.csdn.net/nocml/article/details/125711025 本系列传送门： Transformer(一)–论文翻译：Attention Is All You Need 中文版 Transformer(二)–论文理解：transformer 结构详解 Transformer(三)–论文实… 阅读更多... Spring MVC是一种基于MVC模式的框架，它是Spring框架的一部分。它提供了一种更简单和更有效的方式来构建Web应用程序。它提供了一系列的功能，可以帮助您更快地构建Web应用程序，如控制器，视图，模型，数据绑定&… 阅读更多... 1. 前言之前我们已经详细介绍过如何用AirtestIDE无线连接Android设备： 手把手教你用AirtestIDE无线连接手机！ ，它的关键点在于，需要先 adb connect 一次，才能点击 connect 按钮无线连接上该设备： 但是有很… 阅读更多... 08第六集：LED闪烁第六集：LED闪烁和花式点灯上和花式点灯上.mp4 09第六集：LED闪烁和花式点灯下.mp4 【大文哥学习32位8051】20230704【冲哥视频】第六集的晶振时钟的学习困惑 2023/7/5 17:36 delay_ms(3000); 刷机的时候如果使用11.0592M的Fos… 阅读更多... 文章目录 2211. 统计道路上的碰撞次数（栈 || 脑筋急转弯）解法1：自己想的——使用栈解法2——思维：去掉左右两边往左右开的车代码写法1——找左右端点代码写法2——正则表达式去除流处理api补充：replaceAll() 和正则表… 阅读更多... 🚀写在前面🚀 🖊个人主页：https://blog.csdn.net/m0_52051577?typeblog 🎁欢迎各位大佬支持点赞收藏，三连必回！！ 🔈本人新开系列专栏—python图像处理 ❀愿每一个骤雨初… 阅读更多... 小伙伴们，今天来学习猜数字游戏叭！废话不多说，让我们一起开始学习叭! 思路： 一打开游戏就出现一个菜单然后可以让我们选择是进入游戏还是退出游戏！ #include<stdio.h> void menu() {printf("*****… 阅读更多... 【MySQL】表的增删改查一、插入操作----insert1.1 简单插入1.2 插入时是否更新----ON DUPLICATE KEY UPDATE1.3 插入时替换----REPLACE 二、查询----select2.1 简单查询与去重2.2 基本查询----where条件2.2.3 案列演示 2.4 排序----order by 三、修改操作----update四、删除--… 阅读更多... 前言本篇在讲什么 Lua中的table的实现本篇适合什么适合初学Lua的小白本篇需要什么对Lua语法有简单认知依赖Sublime Text编辑器本篇的特色具有全流程的图文教学重实践，轻理论，快速上手提供全流程的源码内容 ★提高阅读体验★ &amp… 阅读更多... 一、Dump导出案例二、注意事项 MySQL Dump 导出 mysqldump是一个常用的 MySQL 数据库备份工具，它可以将 MySQL 数据库中的数据导出为 SQL 格式的文件，从而实现对数据的备份、迁移和恢复等操作。Doris 在0.15 之后的版本已… 阅读更多... 本文是个人学习笔记，素材来自青岛大学王卓老师的教学视频。一方面用于学习记录与分享，另一方面是想让更多的人看到这么好的《数据结构与算法》的学习视频。如有侵权，请留言作删文处理。课程视频链接： 数据结构与算法基础–… 阅读更多... Redisson 是一种基于 Redis 的 Java 驻留集群的分布式对象和服务库，可以为我们提供丰富的分布式锁和线程安全集合的实现。在 Spring Boot 应用程序中使用 Redisson 可以方便地实现分布式应用程序的某些方面，例如分布式锁、分布式集合、分布式事件发布和订… 阅读更多... Oracle数据库软件安装与卸载实验目的及要求学习Oracle12c数据库服务器软件和客户端软件的安装与卸载,掌握客户端服务名的设置,建立客户端与服务器的网络连接,熟悉windows操作系统中Oracle相关服务的操作。理解数据库管理的基本架构。 （1）熟悉Oracle… 阅读更多... 博主介绍： 大家好，我是一名在Java圈混迹十余年的程序员，精通Java编程语言，同时也熟练掌握微信小程序、Python和Android等技术，能够为大家提供全方位的技术支持和交流。我擅长在JavaWeb、SSH、SSM、SpringBoot等框架… 阅读更多... 情况是这样，我在虚拟机上，使用docker跑前端，需要这个前端支持https，原http的话自动跳转到https。另外，前端部署使用了负载均衡，即使用了3个docker跑前端：1个入口，另外2个是前端&… 阅读更多...