每日提升一点点

cifar10 import os import _pickle as cPickle import numpy as np def load_cifar10(data_dir='../data/'): def load_cifar_batches(filenames): if isinstance(filenames, str): filenames = [filenames] images = [] labels = .

def is_image_file(fileName): ls = [fileName.endswith(extension) for extension in ['.jpg', '.png', '.jpeg']] print(ls) print(any(ls)) fileName = '01.jpg' is_image_file(fileName)

首先，用urllib获取数据，写个压缩包文件，把数据存进去，解压压缩包文件，形成带有数据的目录。 from os.path import exists, basename, join from os import makedirs, remove from six.moves import urllib import tarfile class download_basd300(object): def __init__(self, dest="dataset"): sel.

from os.path import exists, join, basename from os import makedirs, remove from six.moves import urllib import tarfile from torchvision.transforms import Compose, CenterCrop, ToTensor, Resize from dataset import DatasetFromFolder def download_bsd300(de.