【Pytorch｜Dataset】自定义数据集读取（未完结）

文章目录

• 前言
• 常用的库
• • Python 相关库
  • PyTorch 相关库
• 参考代码
• • 参考一

前言

数据集读取就是在做一些文件和路径的 *** 作，多多借鉴网上的一些代码，了解常用的函数，形成自己一个代码套路，可以大大提高自己复现论文的效率。

常用的库 Python 相关库

import os
import json
import random
import cv2
from PIL import Image

PyTorch 相关库

import torch
from torch.utils.data import Dataset
from torchvision import transforms

参考代码 参考一

在pytorch中自定义dataset读取数据

Github链接：视频提到的代码