PyTorch学习笔记（5）——Kaggle猫狗分类问题数据集读取和构建

最新推荐文章于 2025-06-28 22:14:29 发布

原创

最新推荐文章于 2025-06-28 22:14:29 发布 · 814 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #深度学习 #kaggle

Kaggle猫狗分类问题数据集处理

数据集官方下载：https://www.kaggle.com/c/dogs-vs-cats-redux-kernels-edition/data

参考资料：

https://github.com/ytchx1999/Pytorch-Camp
https://github.com/greebear/pytorch-learning

import torch
import os
from PIL import Image
from torch.utils.data import DataLoader
import torchvision.transforms as transforms
import numpy as np
import matplotlib.pyplot as plt

# 数据集（训练集）所在目录
# ../data/Dog_Cat/train

label_mp = {
   
   "cat":0, "dog":1}

# 根据路径获取图片信息
def get_data(data_dir):
    data_dc = []
    for root, dirs, files in os.walk(data_dir):
        for img_name in files:
            img_path = os.path.join(data_dir, img_name)
            label = label_mp[img_name.split('.')[0]]
            data_dc.