|j d dl mZ d dlZd dlZd dlmZmZmZ d dlZ d dl
m Z d dl Z d dl
mZ d dlmZ er(d dlmZ d dlmZ d dlmZ d d
lmZ ed Zed Zg Zd
Z dZ dZ dZ dZ dZ!ddddZ" G d dee#d Z$dS ) ) annotationsN)
TYPE_CHECKINGAnyLiteral)Image)_check_exists_and_download)Dataset) DTypeLike)
_Transform )_ImageDataType)cv2piltrainvalidtestz=https://dataset.bj.bcebos.com/voc/VOCtrainval_11-May-2012.tar 6cd6e144f989b92b3379bac3b3de84fdz/VOCdevkit/VOC2012/ImageSets/Segmentation/{}.txtz#VOCdevkit/VOC2012/JPEGImages/{}.jpgz*VOCdevkit/VOC2012/SegmentationClass/{}.pngvoc2012trainvalr val)r r r c e Zd ZU dZded< ded< ded< ded < d
ed < d ed
< d d dZd Zd dZd!d Zd"d Z dS )#VOC2012a
Implementation of `VOC2012 `_ dataset.
Args:
data_file (str|None, optional): Path to data file, can be set None if
:attr:`download` is True. Default: None, default data path: ~/.cache/paddle/dataset/voc2012.
mode (str, optional): Either train or test mode. Default 'train'.
transform (Callable|None, optional): Transform to perform on image, None for no transform. Default: None.
download (bool, optional): Download dataset automatically if :attr:`data_file` is None. Default: True.
backend (str|None, optional): Specifies which type of image to be returned:
PIL.Image or numpy.ndarray. Should be one of {'pil', 'cv2'}.
If this option is not set, will get backend from :ref:`paddle.vision.get_image_backend `,
default backend is 'pil'. Default: None.
Returns:
:ref:`api_paddle_io_Dataset`. An instance of VOC2012 dataset.
Examples:
.. code-block:: python
>>> # doctest: +TIMEOUT(120)
>>> import itertools
>>> import paddle.vision.transforms as T
>>> from paddle.vision.datasets import VOC2012
>>> voc2012 = VOC2012()
>>> print(len(voc2012))
2913
>>> for i in range(5): # only show first 5 images
... img, label = voc2012[i]
... # do something with img and label
... print(type(img), img.size)
... # (500, 281)
... print(type(label), label.size)
... # (500, 281)
>>> transform = T.Compose(
... [
... T.ToTensor(),
... T.Normalize(
... mean=[0.5, 0.5, 0.5],
... std=[0.5, 0.5, 0.5],
... to_rgb=True,
... ),
... ]
... )
>>> voc2012_test = VOC2012(
... mode="test",
... transform=transform, # apply transform to every image
... backend="cv2", # use OpenCV as image transform backend
... )
>>> print(len(voc2012_test))
1464
>>> for img, label in itertools.islice(iter(voc2012_test), 5): # only show first 5 images
... # do something with img and label
... print(type(img), img.shape) # type: ignore
... # [3, 281, 500]
... print(type(label), label.shape)
... # (281, 500)
str | None data_file _DatasetModemode_Transform[Any, Any] | None transform
_ImageBackendbackendstrflagr
dtypeNr Tdownloadbool_ImageBackend | NonereturnNonec | dv s
J d| | t j }|dvrt d| || _ t | | _ || _ | j 3|s
J d t |t t t | | _ || _
| t j | _ d S )Nr z3mode should be 'train', 'valid' or 'test', but got )r r z4Expected backend are one of ['pil', 'cv2'], but got z>data_file is not set and downloading automatically is disabled)lowerpaddlevisionget_image_backend
ValueErrorr!
MODE_FLAG_MAPr# r r VOC_URLVOC_MD5 CACHE_DIRr
_load_annoget_default_dtyper$ )selfr r r r% r! s n/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/vision/datasets/voc2012.py__init__zVOC2012.__init__ s zz||
H G G
? m5577G . ( ( PwPP
!$**,,/ " > !
P
8 87GY DN #
-//
c i | _ t j | j | _ | j D ]}|| j |j < t | j }| j
| j | }g | _ g | _ |D ]}|
}t | d }t | d }| j | | j | d S )Nzutf-8)name2memtarfileopenr data_tar
getmembersnameSET_FILEformatr# extractfiledatalabelsstrip DATA_FILEdecode
LABEL_FILEappend)r6 eleset_filesetslinerD labels r7 r4 zVOC2012._load_anno s
T^44
=++-- * *C&)DM#( # #??49--}(( x)@AA & &D::< rC r; readr r= ioBytesIOr! nparrayr astyper$ )r6 rP r
label_filerD rO s r7 __getitem__zVOC2012.__getitem__ s IcN [%
}(( y)ABBGGII
))$-
*CDDIIKKz"*T**++
2:e,,-- <5 8D>>DHUOOE > %>>$''D <5 ;;tz**ELL,D,DD DU{r9 c * t | j S N)lenrD r6 s r7 __len__zVOC2012.__len__ s 49~~ r9 c J | j r| j d S d S r] )r> closer_ s r7 __del__zVOC2012.__del__ s0 = " M ! ! ! ! ! " "r9 )Nr NTN) r r r r r r r% r& r! r' r( r) )rP rQ r( rR )r( rQ )r( r) )
__name__
__module__ __qualname____doc____annotations__r8 r4 r[ r` rc r9 r7 r r 6 s A AF ****
III !% $15 (,
%0 %0 %0 %0 %0N& & && * " " " " " "r9 r )r
znpt.NDArray[Any])%
__future__r rU r< typingr r r numpyrW PILr r, paddle.dataset.commonr paddle.ior numpy.typingnptpaddle._typingr
#paddle.vision.transforms.transformsr imager
r r __all__r1 r2 rA rG rI r3 r0 tupler ri r9 r7 rw st # " " " " " . . . . . . . . . .
< < < < < <
5 ((((((>>>>>>&&&&&&L)M34L
I
, < 1
9
$gFF
_" _" _" _" _"ge @AB _" _" _" _" _"r9