CUB200_2011數據集處理

from torch.utils.data import Dataset from PIL import Image import os class CUB_200(Dataset): def __init__(self, root, train=True, transform=None): super(CUB_200, self).__init__()
相關文章
相關標籤/搜索