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

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

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

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

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

在pytorch中自定义dataset读取数据

Github链接：视频提到的代码