kh b d dl Z d dlmZmZmZmZmZ ddlmZ ddl m
Z
ddl m Z G d de Z
y) N)AnyCallableOptionalTupleUnion )default_loader)verify_str_arg)
VisionDatasetc e Zd ZdZddddefdeeej f dede e
de e
d e d
e
ege f d df fd
Z
d efd
Zded ee e f fdZd e fdZd Z xZS ) StanfordCarsa Stanford Cars Dataset
The Cars dataset contains 16,185 images of 196 classes of cars. The data is
split into 8,144 training images and 8,041 testing images, where each class
has been split roughly in a 50-50 split
The original URL is https://ai.stanford.edu/~jkrause/cars/car_dataset.html,
the dataset isn't available online anymore.
.. note::
This class needs `scipy `_ to load target files from `.mat` format.
Args:
root (str or ``pathlib.Path``): Root directory of dataset
split (string, optional): The dataset split, supports ``"train"`` (default) or ``"test"``.
transform (callable, optional): A function/transform that takes in a PIL image or torch.Tensor, depends on the given loader,
and returns a transformed version. E.g, ``transforms.RandomCrop``
target_transform (callable, optional): A function/transform that takes in the
target and transforms it.
download (bool, optional): This parameter exists for backward compatibility but it does not
download the dataset, since the original URL is not available anymore.
loader (callable, optional): A function to load an image given its path.
By default, it uses PIL as its image loader, but users could also pass in
``torchvision.io.decode_image`` for decoding image data into tensors directly.
trainNFrootsplit transformtarget_transformdownloadloaderreturnc | dd l m} t | ||| t
|dd | _ t j | dz | _
| j dz }| j dk( r |d z | _ | j d
z | _ n(| j d z | _ | j d z | _ |r| j | j s t d
|j | j d d D cg c]% } t! | j | d z | d dz
f' c} | _ |j t! |dz d d j% | _ t) | j& D
ci c] \ }
} | |
c} }
| _ || _ y # t $ r t d w xY wc c} w c c} }
w )Nr zQScipy is not found. This dataset needs to have scipy installed: pip install scipy)r r r )r test
stanford_carsdevkitr zcars_train_annos.mat
cars_trainz cars_test_annos_withlabels.mat cars_testzDataset not found.T)
squeeze_me annotationsfnameclassr z
cars_meta.mat class_names)scipy.ioio ImportError RuntimeErrorsuper__init__r
_splitpathlibPath _base_folder_annotations_mat_path_images_base_pathr
_check_existsloadmatstr_samplestolistclasses enumerate class_to_idxr )
selfr r r r r r sior
annotationicls __class__s
^/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/datasets/stanford_cars.pyr& zStanfordCars.__init__& s t " EUV$UG5FG #LL . @""X- ;;' !)/2H)HD &%)%6%6 %ED ")-):):=])]D &%)%6%6%DD " MMO!!# 34 4 "kk$*D*DQUk VWd e
D**Z-@@A7#a'
{{3v'?#@T{RS`ahhj 2;DLL2I J3S!V J A t rs s t,
Ks F >*F3=
F8F0c , t | j S )N)lenr0 r5 s r; __len__zStanfordCars.__len__T s 4==!! idxc | j | \ }}| j | }| j | j | }| j | j | }||fS )z.Returns pil_image and class_id for given index)r0 r r r )r5 rA
image_pathtargetimages r; __getitem__zStanfordCars.__getitem__W sa !]]3 /
F J' >> %NN5)E ,**62Ff} r@ c | j dz j sy| j j xr | j j S )Nr F)r* is_dirr+ existsr, r> s r; r- zStanfordCars._check_existsb sE !!H,446))002Vt7M7M7T7T7VVr@ c t d )NzTThe original URL is broken so the StanfordCars dataset cannot be downloaded anymore.)
ValueErrorr> s r; r zStanfordCars.downloadh s oppr@ )__name__
__module__ __qualname____doc__r r r/ r( r) r r boolr r& intr? r rF r- r
__classcell__)r: s @r; r
r
s < (,/3 '5, C %&, , H% ,
#8, ,
, #$,
, \" " s uS#X Wt W qr@ r
)r( typingr r r r r folderr utilsr
visionr r
r@ r; rX s' 8 8 " ! !_q= _qr@