from sklearn.datasets import load_digits
from torch.utils.data import DataLoader
import numpy as np
digits = load_digits()
img = digits['images'] #(1797,8,8)
img = img[:
sklearn手写体数据集 Dataloader分batch训练
最新推荐文章于 2024-08-16 15:47:12 发布