
    kh                        d Z ddlZddlmZmZ ddlmZ e G d d             Ze G d de             Ze G d	 d
e             Z	e G d d             Z
e G d de
             Ze G d de
             Ze G d de
             Ze G d d             Ze G d de             Ze G d de             Ze G d de             Ze G d de             Ze G d de             Ze G d d e             Zy)!zL COCO, VOC, OpenImages dataset configurations

Copyright 2020 Ross Wightman
    N)	dataclassfield)Dictc                   P    e Zd ZU dZeed<   dZeed<   dZeed<   dZ	e
eef   ed<   y)CocoCfgNvariantcocoparserP   num_classessplits)__name__
__module____qualname__r   str__annotations__r
   r   intr   r   dict     V/var/www/teggl/fontify/venv/lib/python3.12/site-packages/effdet/data/dataset_config.pyr   r   
   s2    GSFCK"FDdO"r   r   c                   D    e Zd ZU dZeed<    ed       Zeee	f   ed<   y)Coco2017Cfg2017r   c            
          t        t        ddd      t        ddd      t        ddd	      t        d
dd	            S )Nz$annotations/instances_train2017.json	train2017Tann_filenameimg_dir
has_labelsz"annotations/instances_val2017.jsonval2017z$annotations/image_info_test2017.jsontest2017Fz(annotations/image_info_test-dev2017.json)trainvaltesttestdevr   r   r   r   <lambda>zCoco2017Cfg.<lambda>   sE    D FP[hlmBIbfgEzfkl"LV`mrs	= r   default_factoryr   N
r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s*    GS# 5 FDdO r   r   c                   D    e Zd ZU dZeed<    ed       Zeee	f   ed<   y)Coco2014Cfg2014r   c            	      f    t        t        ddd      t        ddd      t        ddd	      
      S )Nz$annotations/instances_train2014.json	train2014Tr   z"annotations/instances_val2014.jsonval2014 test2014Fr#   r$   r%   r'   r   r   r   r(   zCoco2014Cfg.<lambda>    s4    D FP[hlmBIbfgr:%H= r   r)   r   Nr+   r   r   r   r-   r-      s*    GS# 5 FDdO r   r-   c                   ^    e Zd ZU dZeed<   dZeed<   dZeed<   dZ	eed<   dZ
eeef   ed	<   y)
VocCfgNr   vocr
   r   r   %s.jpgimg_filenamer   r   r   r   r   r   r   r
   r   r   r9   r   r   r   r   r   r   r6   r6   '   s<    GSFCK L# "FDdO"r   r6   c                   D    e Zd ZU dZeed<    ed       Zeee	f   ed<   y)
Voc2007Cfg2007r   c                  L    t        t        ddd      t        ddd            S )Nz VOC2007/ImageSets/Main/train.txtVOC2007/Annotations/%s.xmlVOC2007/JPEGImagessplit_filenamer   r   zVOC2007/ImageSets/Main/val.txtr#   r$   r'   r   r   r   r(   zVoc2007Cfg.<lambda>3   s/    D=5(, ;5(*
= r   r)   r   Nr+   r   r   r   r<   r<   0   *    GS# 
5 
FDdO 
r   r<   c                   D    e Zd ZU dZeed<    ed       Zeee	f   ed<   y)
Voc2012Cfg2012r   c                  L    t        t        ddd      t        ddd            S )Nz VOC2012/ImageSets/Main/train.txtVOC2012/Annotations/%s.xmlVOC2012/JPEGImagesrA   zVOC2012/ImageSets/Main/val.txtrC   r'   r   r   r   r(   zVoc2012Cfg.<lambda>C   s/    D=5(* ;5(*
= r   r)   r   Nr+   r   r   r   rF   rF   @   rD   r   rF   c                   D    e Zd ZU dZeed<    ed       Zeee	f   ed<   y)
Voc0712Cfg0712r   c                  X    t        t        ddgddgddg      t        ddd      	      S )
Nz#VOC2007/ImageSets/Main/trainval.txtz#VOC2012/ImageSets/Main/trainval.txtr?   rI   r@   rJ   rA   zVOC2007/ImageSets/Main/test.txtrC   r'   r   r   r   r(   zVoc0712Cfg.<lambda>S   sB    DAChi68TU)+?@B <5(*
= r   r)   r   Nr+   r   r   r   rL   rL   P   rD   r   rL   c                   T    e Zd ZU dZeed<   dZeed<   dZeed<   dZ	dZ
eeef   ed<   y)OpenImagesCfgNr   
openimagesr
   r   r8   r   r:   r   r   r   rP   rP   a   s7    GSFCKL"FDdO"r   rP   c                   *    e Zd ZU dZeed<   dZeed<   y)OpenImagesObjCfgiY  r   z*annotations/class-descriptions-boxable.csvcategories_mapNr   r   r   r   r   r   rT   r   r   r   r   rS   rS   j   s    KFNCFr   rS   c                   *    e Zd ZU dZeed<   dZeed<   y)OpenImagesSegCfgi^  r   z$annotations/classes-segmentation.txtrT   NrU   r   r   r   rW   rW   p   s    K@NC@r   rW   c                   6    e Zd ZU  ed       Zeeef   ed<   y)OpenImagesObjV5Cfgc                  x    t        t        dddddd      t        dd	dd
dd      t        dddd
dd            S )Nr#   annotations/train-info.csvT   z&annotations/train-annotations-bbox.csvz;annotations/train-annotations-human-imagelabels-boxable.csvr   img_infor    prefix_levelsann_bboxann_img_label
validationannotations/validation-info.csvr   z+annotations/validation-annotations-bbox.csvz@annotations/validation-annotations-human-imagelabels-boxable.csvr%   r2   z%annotations/test-annotations-bbox.csvz:annotations/test-annotations-human-imagelabels-boxable.csvr4   r'   r   r   r   r(   zOpenImagesObjV5Cfg.<lambda>x   sY    D&Btcd=W

  +LY]mnB\

 RD<V
= r   r)   r   N)	r   r   r   r   r   r   r   r   r   r   r   r   rY   rY   v   s     # 5 FDdO r   rY   c                   R    e Zd ZU dZeed<   dZeed<    ed       Z	e
eef   ed<   y)	OpenImagesObjChallenge2019Cfgi  r   zEannotations/challenge-2019/challenge-2019-classes-description-500.csvrT   c                  x    t        t        dddddd      t        dd	dd
dd      t        ddd
ddd            S )Nr#   r[   Tr\   zBannotations/challenge-2019/challenge-2019-train-detection-bbox.csvzOannotations/challenge-2019/challenge-2019-train-detection-human-imagelabels.csvr]   rb   rc   r   zGannotations/challenge-2019/challenge-2019-validation-detection-bbox.csvzTannotations/challenge-2019/challenge-2019-validation-detection-human-imagelabels.csvchallenge2019z-annotations/challenge-2019/challenge2019-infoFr2   )r   r^   r_   r    r`   ra   r4   r'   r   r   r   r(   z&OpenImagesObjChallenge2019Cfg.<lambda>   sZ    D&BtcdYk

  +LY]mn^p

 #.]mnr
= r   r)   r   N)r   r   r   r   r   r   rT   r   r   r   r   r   r   r   r   re   re      s4    KaNCa# 5 FDdO r   re   c                   D    e Zd ZU dZeed<    ed       Zee	e
f   ed<   y)OpenImagesSegV5Cfg,  r   c                  N    t        t               t               t                     S Nr4   r'   r   r   r   r(   zOpenImagesSegV5Cfg.<lambda>       DfFV= r   r)   r   N)r   r   r   r   r   r   r   r   r   r   r   r   r   r   ri   ri      s*    K# 5 FDdO r   ri   c                   R    e Zd ZU dZeed<   dZeed<    ed       Z	e
eef   ed<   y)	OpenImagesSegChallenge2019Cfgrj   r   zMannotations/challenge-2019/challenge-2019-classes-description-segmentable.csvann_class_mapc                  N    t        t               t               t                     S rl   r'   r   r   r   r(   z&OpenImagesSegChallenge2019Cfg.<lambda>   rm   r   r)   r   N)r   r   r   r   r   r   rp   r   r   r   r   r   r   r   r   ro   ro      s4    KhM3h# 5 FDdO r   ro   )__doc__osdataclassesr   r   typingr   r   r   r-   r6   r<   rF   rL   rP   rS   rW   rY   re   ri   ro   r   r   r   <module>rv      s   
 (  # # # '   '   # # #           # # # G} G G
 A} A A
 )  ( $4  * )   $4  r   