o
V h @ sx d dl mZ d dlZd dlmZ d dlmZmZmZm Z m
Z
ddl m Z ddl
mZmZ ddlmZ G d d
d
eZdS ) ) annotationsN)Path)AnyCallableOptionalTupleUnion )default_loader) download_and_extract_archiveverify_str_arg)
VisionDatasetc s\ e Zd ZdZdZdddddefd" fdd
Zd#ddZd$d d Zd%d d Z d&d d!Z
Z S )' FGVCAircrafta2 `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 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): If True, downloads the dataset from the internet and
puts it in root directory. If dataset is already downloaded, it is not
downloaded again.
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.
zWhttps://www.robots.ox.ac.uk/~vgg/data/fgvc-aircraft/archives/fgvc-aircraft-2013b.tar.gztrainvalvariantNFrootUnion[str, Path]splitstrannotation_level transformOptional[Callable]target_transformdownloadboolloaderCallable[[str], Any]returnNonec s t j|||d t|dd| _t|dd| _tj| jd| _ |r&|
| s.t dtj| j dd d
d d| j }t
|d } d
d | D | _W d n1 sVw Y tt| jtt| j| _tj| j dd}
tj| j dd| j d| j d} g | _g | _t
| d .} | D ]#} | dd\}
}| jtj|
|
d | j| j| qW d n1 sw Y || _d S )N)r r r )trainvalr testr )r family manufacturerzfgvc-aircraft-2013bz;Dataset not found. You can use download=True to download itdataz variants.txtz families.txtzmanufacturers.txtrc S s g | ]}| qS )strip).0liner& r& V/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/datasets/fgvc_aircraft.py
O s z)FGVCAircraft.__init__..imagesimages__z.txt r z.jpg)super__init__r _split_annotation_levelospathjoinr
_data_path _download
_check_exists RuntimeErroropenclassesdictziprangelen class_to_idx _image_files_labelsr' r appendr )selfr r r r r r r annotation_filefimage_data_folder labels_filer)
image_name
label_name __class__r& r* r1 . sH
$
zFGVCAircraft.__init__intc C s
t | jS N)r@ rB rE r& r& r* __len__` s
zFGVCAircraft.__len__idxTuple[Any, Any]c C sH | j | | j| }}| |}| jr| |}| jr | |}||fS rO )rB rC r r r )rE rR
image_filelabelimager& r& r* __getitem__c s
zFGVCAircraft.__getitem__c C s | rdS t| j| j dS )zW
Download the FGVC Aircraft dataset archive and extract it under root.
N)r9 r _URLr rP r& r& r* r8 o s zFGVCAircraft._downloadc C s t j| jo
t j| jS rO )r4 r5 existsr7 isdirrP r& r& r* r9 w s zFGVCAircraft._check_exists)r r r r r r r r r r r r r r r r )r rN )rR rN r rS )r r )r r ) __name__
__module__ __qualname____doc__rX r
r1 rQ rW r8 r9
__classcell__r& r& rL r* r
s
2
r )
__future__r r4 pathlibr typingr r r r r folderr
utilsr r visionr
r r& r& r& r* s