,}h z d dl mZ d dlZd dlmZ d dlmZmZmZm Z m
Z
d dl Z ddl
mZmZ ddlmZ G d d e Zy)
) annotationsN)Path)AnyCallableOptionalTupleUnion ) download_and_extract_archiveverify_str_arg)
VisionDatasetc n e Zd ZdZdZ d d fd
Zd
dZd dZd dZd
dZ xZ
S ) FGVCAircrafta `FGVC Aircraft `_ Dataset.
The dataset contains 10,000 images of aircraft, with 100 images for each of 100
different aircraft model variants, most of which are airplanes.
Aircraft models are organized in a three-levels hierarchy. The three levels, from
finer to coarser, are:
- ``variant``, e.g. Boeing 737-700. A variant collapses all the models that are visually
indistinguishable into one class. The dataset comprises 100 different variants.
- ``family``, e.g. Boeing 737. The dataset comprises 70 different families.
- ``manufacturer``, e.g. Boeing. The dataset comprises 30 different manufacturers.
Args:
root (str or ``pathlib.Path``): Root directory of the FGVC Aircraft dataset.
split (string, optional): The dataset split, supports ``train``, ``val``,
``trainval`` and ``test``.
annotation_level (str, optional): The annotation level, supports ``variant``,
``family`` and ``manufacturer``.
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.
download (bool, optional): If True, downloads the dataset from the internet and
puts it in root directory. If dataset is already downloaded, it is not
downloaded again.
zWhttps://www.robots.ox.ac.uk/~vgg/data/fgvc-aircraft/archives/fgvc-aircraft-2013b.tar.gzc
t | ||| t |dd | _ t |dd | _ t
j j | j d | _ |r| j | j s t d t
j j | j dd d
d d| j }t |d 5 }|D cg c] } | j c} | _ d d d t! t# | j t% t' | j | _ t
j j | j dd
}
t
j j | j dd| j d| j d } g | _ g | _ t | d 5 }|D ] } | j j/ dd \ } }
| j* j1 t
j j |
| d | j, j1 | j( |
d d d y c c} w # 1 sw Y sxY w# 1 sw Y y xY w)N) transformtarget_transformsplit)trainvaltrainvaltestannotation_level)variantfamily manufacturerzfgvc-aircraft-2013bz;Dataset not found. You can use download=True to download itdataz variants.txtz families.txtzmanufacturers.txtrimagesimages__z.txt r
z.jpg)super__init__r _split_annotation_levelospathjoinroot
_data_path _download
_check_exists RuntimeErroropenstripclassesdictziprangelen class_to_idx _image_files_labelsr append)selfr) r r r r downloadannotation_fileflineimage_data_folder labels_file
image_name
label_name __class__s a/var/www/html/test/engine/venv/lib/python3.12/site-packages/torchvision/datasets/fgvc_aircraft.pyr# zFGVCAircraft.__init__+ s+ EUV$UG5YZ !/ 02W"
'',,tyy2GH NN !!# \] ] '',, OO )( 3
$$
&
/3
' 81567TDJJL7DL 8 !T\\5T\\9J3K!L M GGLL &( Kggll4??FgdF\F\E]]^_c_j_j^kko,I11I;>Jc , t | j S N)r4 r6 r9 s rC __len__zFGVCAircraft.__len__[ s 4$$%% c | j | | j | }}t j j | j d }| j r| j
| }| j r| j | }||fS )NRGB)r6 r7 PILImager. convertr r )r9 idx
image_filelabelimages rC __getitem__zFGVCAircraft.__getitem__^ st - -c 2DLL4EE
z*2259 >>NN5)E ))%0Ee|rH c f | j ryt | j | j y)zW
Download the FGVC Aircraft dataset archive and extract it under root.
N)r, r _URLr) rF s rC r+ zFGVCAircraft._downloadj s% $TYY :rH c t j j | j xr) t j j | j S rE )r& r' existsr* isdirrF s rC r, zFGVCAircraft._check_existsr s/ ww~~doo.Q277== 3QQrH )r r NNF)r) zUnion[str, Path]r strr rX r Optional[Callable]r rY r: boolreturnNone)r[ int)rN r] r[ zTuple[Any, Any])r[ r\ )r[ rZ ) __name__
__module__ __qualname____doc__rT r# rG rR r+ r,
__classcell__)rB s @rC r r
s 6 eD
)(,/3 .C .C .C .C
& .C -
.C .C
.C`&
;RrH r )
__future__r r& pathlibr typingr r r r r PIL.ImagerK utilsr r visionr
r rH rC rj s- " 8 8 ? !fR= fRrH