1.数据集文件夹

train文件夹（44229张），test文件夹（4741张），valid文件夹（6000张）

2.train文件夹展示

 

labels展示

 

标签txt展示

data.yaml文件展示 

 

 对数据集感兴趣的可以关注最后一行

import os
import json

import torch
from PIL import Image
from torchvision import transforms
import tkinter as tk
from tkinter import filedialog

from model import CNN
from PIL import ImageTk

#数据集https://mbd.pub/o/bread/ZJ2YlZ5p