a
PSicX
@ st d dl Z d dlZd dlmZmZmZmZmZ d dlm Z d dl
m Z ddl m
Z
mZ ddlmZ G dd d eZdS )
N)AnyCallableOptionalTupleList)urlparse)Image ) download_and_extract_archiveverify_str_arg)
VisionDatasetc s e Zd ZdZdZdZdeeee ee e dd fdd
Z
e d
d d Z e e
eef d
ddZe d
ddZdd
ddZed
ddZ ZS )CLEVRClassificationap `CLEVR `_ classification dataset.
The number of objects in a scene are used as label.
Args:
root (string): Root directory of dataset where directory ``root/clevr`` exists or will be saved to if download is
set to True.
split (string, optional): The dataset split, supports ``"train"`` (default), ``"val"``, or ``"test"``.
transform (callable, optional): A function/transform that takes in an PIL image and returns a transformed
version. E.g, ``transforms.RandomCrop``
target_transform (callable, optional): A function/transform that takes in them 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.
z3https://dl.fbaipublicfiles.com/clevr/CLEVR_v1.0.zipZ b11922020e72d0cd9154779b2d3d07d2trainNF)rootsplit transformtarget_transformdownloadreturnc s t |dd| _t j|||d t| jd | _| jtt| j j
j | _ |rZ|
| sjtdt| j d| jd| _| | jdkrt| j d d
| j d }t|}W d n1 s0 Y d d
|d D fdd| jD | _nd gt| j | _d S )Nr )r valtest)r r clevrzHDataset not found or corrupted. You can use download=True to download itimages*r ZscenesZCLEVR_z _scenes.jsonc S s i | ]}|d t |d qS )Zimage_filenameobjects)len).0scene r V/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/datasets/clevr.py
8 z0CLEVRClassification.__init__..c s g | ]} |j qS r )name)r
image_file num_objectsr r
9 r! z0CLEVRClassification.__init__..)r _splitsuper__init__pathlibPathr _base_folderr _URLpathstem _data_folder _download
_check_exists RuntimeErrorsortedjoinpathglob _image_filesopenjsonload_labelsr )selfr r r r r filecontent __class__r$ r r) s (z CLEVRClassification.__init__)r c C s
t | jS N)r r7 r< r r r __len__= s zCLEVRClassification.__len__)idxr c C sL | j | }| j| }t|d}| jr4| |}| jrD| |}||fS )NRGB)r7 r; r r8 convertr r )r< rD r# labelimager r r __getitem__@ s
z CLEVRClassification.__getitem__c C s | j o| j S rA )r0 existsis_dirrB r r r r2 N s z!CLEVRClassification._check_existsc C s( | r d S t| jt| j| jd d S )N)md5)r2 r
r- strr, _MD5rB r r r r1 Q s z CLEVRClassification._downloadc C s d| j S )Nzsplit=)r' rB r r r
extra_reprW s z CLEVRClassification.extra_repr)r NNF)__name__
__module__ __qualname____doc__r- rN rM r r boolr) intrC r r rI r2 r1 rO
__classcell__r r r? r r
s( r
)r9 r* typingr r r r r urllib.parser PILr utilsr
r visionr r
r r r r s