「小土堆」pytorch DataSet
「小土堆」pytorch DataSet from cProfile import labelfrom torch.utils.data import Dataset from PIL import Image import osclass MyData(Dataset):def __init__(self, root_dir, label_dir):# root_dir "hymenoptera_data/train"# label_dir "ants_img…
2025-10-16