【Pytorch图像分类训练DAY1】构建自己的图像分类数据集

最新推荐文章于 2023-05-05 19:40:17 发布

vegetable6

最新推荐文章于 2023-05-05 19:40:17 发布

阅读量225

点赞数

文章标签： pytorch 分类 python

本文链接：https://blog.csdn.net/vegetable6/article/details/128706225

版权

这篇博客介绍了如何使用Pytorch构建一个81类水果的图像分类数据集。首先，通过云GPU平台下载并解压数据集，接着删除不需要的文件如__.MACOSX__, .DS_Store和.ipynb_checkpoints。然后，利用OpenCV和PIL库检查并删除非三通道图像和读取错误的文件。最后，文章提到了划分训练集和测试集以及可视化的步骤。" 135375939,7337247,GANs与人脸识别技术的融合应用,"['人工智能', '深度学习', '图像处理', '生成对抗网络', '人脸识别']

摘要由CSDN通过智能技术生成

https://github.com/TommyZihao/Train_Custom_Dataset
81类的水果数据集
训练集
测试集都有81个文件夹

用云GPU平台

pip install numpy pandas matplotlib requests tqdm opencv-python

删除多余文件

导入工具包

import os
import cv2
from tqdm import tqdm

准备样例数据集

下载测试数据集压缩包

!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/dataset/dataset_delete_test.zip

–2022-08-02 12:57:00-- https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/dataset/dataset_delete_test.zip
Connecting to 172.16.0.13:5848… connected.
Proxy request sent, awaiting response… 200 OK
Length: 29954078 (29M) [application/zip]
Saving to: ‘dataset_delete_test.zip’

dataset_delete_test 100%[===========