构建自己的图像分类数据集
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%[===========