1.数据集文件夹
train文件夹(44229张),test文件夹(4741张),valid文件夹(6000张)

2.train文件夹展示
 
 
labels展示
 
 
标签txt展示

data.yaml文件展示

对数据集感兴趣的可以关注最后一行
import os
import jsonimport torch
from PIL import Image
from torchvision import transforms
import tkinter as tk
from tkinter import filedialogfrom model import CNN
from PIL import ImageTk#数据集https://mbd.pub/o/bread/ZJ2YlZ5p