
    khJ                     \    d Z ddlZddlmZ ddlmZ ddl ddl ddlm	Z	 ddlm
Z
 d
d	Zy)z/ Dataset factory

Copyright 2020 Ross Wightman
    N)OrderedDict)Path   )*)DetectionDatset)create_parserc                    t        |t              r|f}| j                         } t        |      }t        }t               }| j                  d      rd| v rt               }n
t               }|D ]v  }||j                  vrt        | d      |j                  |   }||d   z  }t        ||d         }	 ||t        |d         z  t        |j                  |	      	      ||<   x n| j                  d
      rd| v rt               }nd| v rt               }n
t!               }|D ]  }||j                  vrt        | d      |j                  |   }t        |d   t"        t$        f      rt'        |d         t'        |d         k(  sJ d }
t)        |d   |d   |d         D ]  \  }}}t+        ||z  t,        j.                  j1                  ||      t,        j.                  j1                  ||j2                              }	|
t        |j                  |	      }
xt        |j                  |	      }|
j5                  |        nwt+        ||d   z  t,        j.                  j1                  ||d         t,        j.                  j1                  |d   |j2                              }	t        |j                  |	      }
 |||
	      ||<    n| j                  d      rd| v rt7               }n
t9               }|D ]  }||j                  vrt        | d      |j                  |   }t;        ||j<                  z  ||d   z  ||d   z  ||d   z  |j2                  |d   |d         }	 ||t        |d         z  t        |j                  |	      	      ||<    nJ d|  d       t%        |j?                               }t'        |      dkD  r|S |d   S )Ncocococo2014z split not found in configann_filename
has_labels)r   r   img_dir)cfg)data_dirparservocvoc0712voc2007split_filename)r   r   img_filename)other
openimageschallenge2019img_infoann_bboxann_img_labelprefix_levels)categories_filenameimg_info_filenamebbox_filenameimg_label_filenamer   r   r   zUnknown dataset parser ()r   r   ) 
isinstancestrlowerr   r   r   
startswithCoco2014CfgCoco2017CfgsplitsRuntimeErrorCocoParserCfgr   r   
Voc0712Cfg
Voc2007Cfg
Voc2012CfgtuplelistlenzipVocParserCfgospathjoinr   mergeOpenImagesObjChallenge2019CfgOpenImagesObjV5CfgOpenImagesParserCfgcategories_mapvalues)namerootr)   dataset_clsdatasetsdataset_cfgs	split_cfgann_file
parser_cfgr   sfafidother_parsers                  W/var/www/teggl/fontify/venv/lib/python3.12/site-packages/effdet/data/dataset_factory.pycreate_datasetrK      s   &#::<D:D!K}Hv%-K%-K 	A***"aS(B#CDD#**1-Ii77H&%$\2J &Yy%9 ::$[%7%7ZHHQK	 
	$,K$$,K$,K 	DA***"aS(B#CDD#**1-I)$45t}E9%5673y?X;YYYY"%!"23Y~5NPYZcPd#f 
9JBB!-'+by%'WW\\$%;%'WW\\"k6N6N%O"QJ ~!.{/A/Az!R'4[5G5GZ'X<8
9 *#')4D*E#E!#dIn4M!N!#i	.BKD\D\!]

 '{'9'9zJ%tFCHQK3	D4 
	&d"79K,.K 	A***"aS(B#CDD#**1-I,$(;+E+E$E"&:)>">"Yz%::#')O*D#D(55'8$\2J &Yy%9 ::$[%7%7ZHHQK	$ 	90a88uHOO%&H8}q(89hqk9    ))trainval)__doc__r4   collectionsr   pathlibr   dataset_configparsersdatasetr   r   rK    rL   rJ   <module>rV      s'    
 #    $ "T:rL   