j t d dl Z d dlmZ d dlZd dlZd dlmZ ddlmZ ddl m
Z
m Z dgZ G d de Z
dS ) N)Any)tqdm )
VisionDataset)convert_target_to_relativecrop_bboxes_from_imageSVHNc
Z e Zd ZdZdZdZ ddededed ed
ed d f fd
Zd e fdZ
xZ S )r a SVHN dataset from `"The Street View House Numbers (SVHN) Dataset"
`_.
.. image:: https://doctr-static.mindee.com/models?id=v0.5.0/svhn-grid.png&src=0
:align: center
>>> from doctr.datasets import SVHN
>>> train_set = SVHN(train=True, download=True)
>>> img, target = train_set[0]
Args:
train: whether the subset should be the training one
use_polygons: whether polygons should be considered as rotated bounding box (instead of straight ones)
recognition_task: whether the dataset should be used for recognition task
detection_task: whether the dataset should be used for detection task
**kwargs: keyword arguments from `VisionDataset`.
)z3http://ufldl.stanford.edu/housenumbers/train.tar.gz@4b17bb33b6cd8f963493168f80143da956f28ec406cc12f8e5745a9f91a51898zsvhn_train.tar)z2http://ufldl.stanford.edu/housenumbers/test.tar.gz@57ac9ceb530e4aa85b55d991be8fc49c695b3d71c6f6a88afea86549efde7fb5z
svhn_test.tarTFtrain use_polygonsrecognition_taskdetection_taskkwargsreturnNc |r| j n| j \ }}} t j |f||d|st nd d| |r|rt d || _ g | _ t j } t j | j
|rdnd }
t j t j |
d d 5 d } d } t! t# | | d
t% | D ]\ }
}d t' t( |
d
d }t j t j |
| s0t/ dt j |
| |d
}|d j d
dk r d | D }n fd| D }t j |d |d |d |d g| }t9 t' t: |d }|rt j t j |d d d
f |d d df gd t j |d d d
f |d d df z |d d df gd t j |d d d
f |d d df z |d d df |d d d f z gd t j |d d d
f |d d df |d d d f z gd gd }n\t j |d d d
f |d d df |d d d
f |d d df z |d d df |d d d f z gd }|rt? t j |
| | }t# || D ]Z\ }}|j d
d
k rD|j d d
k r3t% | d
k r d |vr | j ||f [|r | j ||f | j |tC || f d d d n # 1 swxY w Y |
| _
d S ) NT) file_name file_hashextract_archivepre_transformsz`recognition_task` and `detection_task` cannot be set to True simultaneously. To get the whole dataset with boxes and labels leave both parameters to False.r
testzdigitStruct.matrzdigitStruct/namezdigitStruct/bboxzPreparing and Loading SVHN)iterabledesctotal r zunable to locate leftr c N i | ]"\ }}|t |d d g#S )r )int).0kvalss V/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/datasets/svhn.py
z!SVHN.__init__..` s/ P P PDCQ
OO#4 P P P c 4 i | ]\ }}|fd |D S )c v g | ]5}t |d d 6S )r r )r! item)r" vfs r%
z,SVHN.__init__...b s9 #J#J#JC!A$ (8(8(:(:$;$;#J#J#Jr' r )r" r# r$ r, s r% r& z!SVHN.__init__..b s5 f f fwqRV#J#J#J#JT#J#J#J f f fr' topwidthheight)dtypelabel)axis )img_pathgeoms )boxeslabels)"TRAINTESTsuper__init__r
ValueErrorr
datanpfloat32ospathjoinrooth5pyFiler ziplenmapchrflattenexistsFileNotFoundErrorshapeitemsarray transposeliststrstackr appenddict)selfr
r r r r urlsha256namenp_dtypetmp_rootimg_refsbox_refsimg_refbox_refimg_nameboxbox_dictcoords
label_targets box_targetscropscropr2 r, __class__s @r% r? z
SVHN.__init__2 s +0 >DJJTYVT
=MW55SW
c
VX :7<< e+G77HH Yrw||H.?@@#
F
F 9 `!+,H+,H$( Xx007S[^_g[h[h% % % 6
` 6
` 773sAgajM",=,E,E,G,G#H#HII w~~bgll8X&F&FGG d+,b XW_@`@`,b,bccc
mv;$Q'1,, P PCIIKK P P PHH f f f fZ]ZcZcZeZe f f fH &(X f%x8I8T\K]^fn& & &)++ !%Shw.?%@%@ A A
.0h HfQQQTlF111a4L%A K K K HfQQQTlVAAAqD\&A6!!!Q$<%PWY Z Z Z HfQQQTlVAAAqD\&A6!!!Q$rz s
# # # # # # E E E E E E E E ({% {% {% {% {%= {% {% {% {% {%r'