o * i@sdZddlZddlZddlZddlZddlmZgZdZ e dZ dZ e dZ dZ dd d Zed d dddddZed d dddddZed dddddddZed dddddddZed ddddddZdS)a, CIFAR dataset. This module will download dataset from https://dataset.bj.bcebos.com/cifar/cifar-10-python.tar.gz and https://dataset.bj.bcebos.com/cifar/cifar-100-python.tar.gz, parse train/test set into paddle reader creators. The CIFAR-10 dataset consists of 60000 32x32 color images in 10 classes, with 6000 images per class. There are 50000 training images and 10000 test images. The CIFAR-100 dataset is just like the CIFAR-10, except it has 100 classes containing 600 images each. There are 500 training images and 100 testing images per class. N) deprecatedz$https://dataset.bj.bcebos.com/cifar/zcifar-10-python.tar.gzZ c58f30108f718f92721af3b95e74349azcifar-100-python.tar.gzZ eb9058c3a382ffc7106e4002c42a8d85Fcsddfdd}|S)Ncss\|d}|d|dd}|dusJt||D]\}}|dtjt|fVqdS)Nsdataslabelss fine_labelsgo@)getzipZastypenumpyZfloat32int)batchdatalabelssamplelabelr `/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddle/dataset/cifar.py read_batch0s z"reader_creator..read_batchc3sz tjdd'}fdd|D}|D]}tj||dd}|EdHqWdn1s3wYs:s z1reader_creator..reader..bytes)encoding)tarfileopenpickleload extractfile)fnamesrrcyclefilenamerrr r reader7s  zreader_creator..readerr )r rrr!r rr reader_creator/sr"z2.0.0zpaddle.vision.datasets.Cifar100z>Please use new dataset API which supports paddle.io.DataLoader)ZsinceZ update_tolevelreasoncCttjjtdtdS)z CIFAR-100 training set creator. It returns a reader creator, each sample in the reader is image pixels in [0, 1] and label in [0, 99]. :return: Training reader creator :rtype: callable cifartrainr"paddledatasetcommondownload CIFAR100_URL CIFAR100_MD5r r r r train100Jr0cCr&)z CIFAR-100 test set creator. It returns a reader creator, each sample in the reader is image pixels in [0, 1] and label in [0, 99]. :return: Test reader creator. :rtype: callable r'testr)r r r r test100`r1r3zpaddle.vision.datasets.Cifar10cCttjjtdtd|dS)a CIFAR-10 training set creator. It returns a reader creator, each sample in the reader is image pixels in [0, 1] and label in [0, 9]. :param cycle: whether to cycle through the dataset :type cycle: bool :return: Training reader creator :rtype: callable r'Z data_batchrr"r*r+r,r- CIFAR10_URL CIFAR10_MD5r5r r r train10v r9cCr4)a CIFAR-10 test set creator. It returns a reader creator, each sample in the reader is image pixels in [0, 1] and label in [0, 9]. :param cycle: whether to cycle through the dataset :type cycle: bool :return: Test reader creator. :rtype: callable r'Z test_batchr5r6r5r r r test10r:r;cCs(tjjtdttjjtdtdS)Nr')r*r+r,r-r7r8r.r/r r r r fetchsr<)F)__doc__rrrZpaddle.dataset.commonr*Z paddle.utilsr__all__Z URL_PREFIXr7r8r.r/r"r0r3r9r;r<r r r r s`