机器学习笔记 - 基于pytorch的自定义数据集和数据加载器

server/2024/9/23 14:27:52/

         PyTorch 提供了torch.utils.data.DataLoadertorch.utils.data.Dataset 允许您使用预加载的数据集以及您自己的数据。 Dataset存储样本及其相应的标签,并基于DataLoader进行迭代Dataset以访问样本。

        自定义 Dataset 类必须实现三个函数:__init__、__len__和__getitem__。

        下面的代码,是基于voc格式的目标检测的数据集编写的。

        __init__函数里面定义了基本信息,我们可以看到有路径、图像宽高、分类信息、图像后缀等,主要的是所有图像的路径信息的集合。

        __len__最简单,就是所有的图像的数量。

        __getitem__是根据给定索引找到图像,进行处理,并继续找到对应目标检测的xml文件,读取解析里面的坐标信息等


http://www.ppmy.cn/server/23868.html

相关文章

leetCode59. 螺旋矩阵 II

leetCode59. 螺旋矩阵 II 题目思路&#xff1a;见我的这篇博客 代码 class Solution { public:vector<vector<int>> generateMatrix(int n) {// n * n的矩阵的初始化vector<vector<int>> res(n, vector<int> (n));// 方向数组int dx[] {0, 1…

【RAG 论文】Chain-of-Note:为 RAG 引入 CoT 让模型生成阅读笔记来提高面对噪音文档和未知场景的鲁棒性

论文&#xff1a;Chain-of-Note: Enhancing Robustness in Retrieval-Augmented Language Models ⭐⭐⭐ Tencent AI Lab, arXiv:2311.09210 文章目录 一、论文速读二、实现的细节2.1 Note Design2.2 Data Collection2.3 Model Training 三、实验结果3.1 QA Performance3.2 对 …

vue3前端请求后端接口动态渲染菜单

//获取数据 请求接口 export function parkEnterPrise(address: string, methods: string) { const res instance({ url: address, method: methods, }); return res; } //页面 <el-menu default-active"2" class"el-menu-vertical-demo" …

电机控制器电路板布局布线参考指导(六)功率器件布局布线

电机控制器电路板布局布线参考指导&#xff08;六&#xff09;功率器件布局布线 1.MOS管常见的封装1.1 DPAK1.2 D2PAK1.3 TO-2201.4 引脚 SON 2.MOS管布局3.功率器件布局设计3.1 开关节点3.2 大电流环路路径3.3 VDRAIN 感测引脚 栅极驱动器和功率 MOSFET 的放置对于前置驱动器电…

游戏发行困境及OgGame云游戏解决方案简述

随着全球化浪潮的持续推进&#xff0c;中国游戏开发者们不再满足于国内市场的发展&#xff0c;而是开始将目光投向更为广阔的海外市场。这一趋势的崛起背后&#xff0c;是中国企业意识到国际化是其发展的必由之路&#xff0c;也是游戏行业突破国内困境的体现。本文将简要阐述游…

【数据结构初阶】时间复杂度和空间复杂度详解

今天我们来详细讲讲时间复杂度和空间复杂度&#xff0c;途中如果有不懂的地方可翻阅我之前文章。 个人主页&#xff1a;小八哥向前冲~-CSDN博客 数据结构专栏&#xff1a;数据结构【c语言版】_小八哥向前冲~的博客-CSDN博客 c语言专栏&#xff1a;c语言_小八哥向前冲~的博客-CS…

基于SpringBoot+Vue高校汉服租赁网站的设计与实现

项目介绍&#xff1a; 高校汉服租赁网站管理系统按照操作主体分为管理员和用户。管理员的功能包括字典管理、交流论坛管理、公告资讯管理、汉服信息管理、汉服收藏管理、汉服评价管理、汉服租赁管理、用户管理、管理员管理。用户的功能等。该系统采用了Mysql数据库&#xff0c…

阿里开源黑白图片上色算法DDColor的部署与测试并将模型转onnx后用c++推理

阿里开源黑白图片上色算法DDColor的部署与测试并将模型转onnx后用c推理 文章目录 阿里开源黑白图片上色算法DDColor的部署与测试并将模型转onnx后用c推理简介环境部署下载源码安装环境下载模型 测试一下看看效果模型转onnx测试一下生成的onnx模型看看效果C 推理 简介 DDColor是…