Di NddlmZmZmZddlZddlmZddlmZGddeZ y) )AnyCallableOptionalN) transforms) VisionDatasetceZdZdZ ddedeeeefdedeedeeded dffd Zd ed ee e ffd Z d efd Z xZ S)FakeDataaA fake dataset that returns randomly generated images and returns them as PIL images Args: size (int, optional): Size of the dataset. Default: 1000 images image_size(tuple, optional): Size of the returned images. Default: (3, 224, 224) num_classes(int, optional): Number of classes in the dataset. Default: 10 transform (callable, optional): A function/transform that takes in a PIL image and returns a transformed version. E.g, ``transforms.RandomCrop`` target_transform (callable, optional): A function/transform that takes in the target and transforms it. random_offset (int): Offsets the index-based random seed used to generate each image. Default: 0 Nsize image_size num_classes transformtarget_transform random_offsetreturnc`t|||||_||_||_||_y)N)rr)super__init__r rr r)selfr r rrrr __class__s _/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/torchvision/datasets/fakedata.pyrzFakeData.__init__s6 9?OP &$*indexcd|t|k\r"t|jjdt j }t j ||jzt j|j}t jd|jdtjd}t j|tj|}|j |j!|}|j"|j#|}||j%fS)z Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class. z index out of ranger)r)r dtype)len IndexErrorr__name__torch get_rng_state manual_seedrrandnr randintrlong set_rng_stater ToPILImagerritem)rr rng_stateimgtargets r __getitem__zFakeData.__getitem__(s CI  7 788KLM M'')  %$"4"445kk4??+q$"2"2UZZPQRS I&&j##%c* >> %..%C  ,**62FFKKM!!rc|jS)N)r )rs r__len__zFakeData.__len__Bs yyr)i)r0 NNr) r __module__ __qualname____doc__inttuplerrrrr,r. __classcell__)rs@rr r s "+8(,/3 + +#sC-( + + H% + #8, + +  +""sCx"4rr ) typingrrrr rvisionr r rrr<s ** !:}:r