j d dl Z d dlZd dlmZ d dlmZ d dlZd dlmZ ddl m
Z
ddl m Z m
Z
dgZ G d de
ZdS )
N)Path)Any)tqdm )AbstractDataset)convert_target_to_relativecrop_bboxes_from_imageIC13c H e Zd ZdZ d dedededededed d
f fd
Z xZS )
r
a IC13 dataset from `"ICDAR 2013 Robust Reading Competition" `_.
.. image:: https://doctr-static.mindee.com/models?id=v0.5.0/ic13-grid.png&src=0
:align: center
>>> # NOTE: You need to download both image and label parts from Focused Scene Text challenge Task2.1 2013-2015.
>>> from doctr.datasets import IC13
>>> train_set = IC13(img_folder="/path/to/Challenge2_Training_Task12_Images",
>>> label_folder="/path/to/Challenge2_Training_Task1_GT")
>>> img, target = train_set[0]
>>> test_set = IC13(img_folder="/path/to/Challenge2_Test_Task12_Images",
>>> label_folder="/path/to/Challenge2_Test_Task1_GT")
>>> img, target = test_set[0]
Args:
img_folder: folder with all the images of the dataset
label_folder: folder with all annotation files for the images
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 `AbstractDataset`.
F
img_folder label_folder use_polygonsrecognition_taskdetection_taskkwargsreturnNc t j |fd|st nd i| |r|rt d t j | r t j | s3t dt j | s|n| g | _ t j
}t j | }t |dt | D ]i} t || }
t |dt | j z dz } t! | d 5 } d
t# j | d d
D }
d d d n # 1 swxY w Y d |
D }t j d |
D | }|r t j d |D | }|rIt) |
| }t+ || D ]%\ }}d |vr | j ||f & |r | j |
|f >| j |
t/ || f kd S )Npre_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.zunable to locate zPreparing and Loading IC13)iterabledesctotalgt_z.txt
)newlinec & g | ]}d |D S )c P g | ]#}| d r
|dd n|$S ),N)endswith).0vals V/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/datasets/ic13.py
z,IC13.__init__...O s5 KKK c!2!2;S"XXKKK )r rows r" r# z!IC13.__init__..N s7 LKsKKK r$ ') delimiter quotecharc F g | ] }|d dd S )r " )replacer lines r" r# z!IC13.__init__..R s, CCCDd2h&&sB//CCCr$ c
b g | ],}t t t |d d -S )N )listmapintr/ s r" r# z!IC13.__init__..T s2 /\/\/\TSd2A2h5G5G0H0H/\/\/\r$ )dtypec g | ]@}|d |d g|d |d g|d |d g|d |d ggAS )r r r% )r coordss r" r# z!IC13.__init__..X sl # $AYq 2 #AYq 2 #AYq 2 #AYq 2 r$ )img_pathgeoms)boxeslabels)super__init__r
ValueErrorospathexistsFileNotFoundErrordatanpfloat32listdirr lenr stemopencsvreaderarrayr zipappenddict)selfr r
r r r r np_dtype img_namesimg_namer;
label_pathf_linesr> box_targetscropscroplabel __class__s r" r@ z
IC13.__init__, s"
IY'c'A'A_c
gm
c
w~~l++ 27>>*3M3M #f|8T8T$dLLZdff
XZ :Jz** i6RZ]^gZhZhiii # U # UHJ11H lEDNN4G,G&,PQQJj$///
1 "z!sc J J J
DCFCCCF&(h/\/\U[/\/\/\dl&m&m&mK
h '2 #
U. TTT#&uf#5#5 8 8KD%%'' (($ 7778
U (K!89999 (D{6,R,R,R!STTTTG# U # Us ="E++E/ 2E/ )FFF) __name__
__module__ __qualname____doc__strboolr r@
__classcell__)r^ s @r" r
r
s 6 #!& $
@U @U@U @U @U
@U
@U @U
@U @U @U @U @U @U @U @U @U @Ur$ )rM rB pathlibr typingr numpyrG r datasetsr utilsr r __all__r
r% r$ r" rl s
% % % % % % E E E E E E E E (XU XU XU XU XU? XU XU XU XU XUr$