
    Ng                        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dS )!zL COCO, VOC, OpenImages dataset configurations

Copyright 2020 Ross Wightman
    N)	dataclassfield)Dictc                   X    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<   dS )CocoCfgNvariantcocoparserP   num_classessplits)__name__
__module____qualname__r   str__annotations__r
   r   intr   r   dict     V/var/www/html/ai-engine/env/lib/python3.11/site-packages/effdet/data/dataset_config.pyr   r   
   sZ         GSFCK"FDdO"""""r   r   c                   R    e Zd ZU dZeed<    ed           Zeee	f         ed<   dS )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>   sg    D FP[hlmmmBIbfgggEzfklll"LV`mrsss	= = = r   default_factoryr   N
r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s_         GS#e 5 5   FDdO     r   r   c                   R    e Zd ZU dZeed<    ed           Zeee	f         ed<   dS )Coco2014Cfg2014r   c            	          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>    sP    D FP[hlmmmBIbfgggr:%HHH= = = r   r)   r   Nr+   r   r   r   r-   r-      s_         GS#e 5 5   FDdO     r   r-   c                   f    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	<   dS )
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   '   sj         GSFCK L#   "FDdO"""""r   r6   c                   R    e Zd ZU dZeed<    ed           Zeee	f         ed<   dS )
Voc2007Cfg2007r   c                  d    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   sM    D=5(, , , ;5(* * *
= 
= 
= r   r)   r   Nr+   r   r   r   r<   r<   0   _         GS#e 
5 
5 
 
 
FDdO 
 
 
 
 
r   r<   c                   R    e Zd ZU dZeed<    ed           Zeee	f         ed<   dS )
Voc2012Cfg2012r   c                  d    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   sM    D=5(* * * ;5(* * *
= 
= 
= r   r)   r   Nr+   r   r   r   rF   rF   @   rD   r   rF   c                   R    e Zd ZU dZeed<    ed           Zeee	f         ed<   dS )
Voc0712Cfg0712r   c                  p    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 B B <5(* * *
= 
= 
= r   r)   r   Nr+   r   r   r   rL   rL   P   rD   r   rL   c                   \    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<   dS )OpenImagesCfgNr   
openimagesr
   r   r8   r   r:   r   r   r   rP   rP   a   s_         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<   dS )OpenImagesObjCfgiY  r   z*annotations/class-descriptions-boxable.csvcategories_mapNr   r   r   r   r   r   rT   r   r   r   r   rS   rS   j   s2         KFNCFFFFFr   rS   c                   ,    e Zd ZU dZeed<   dZeed<   dS )OpenImagesSegCfgi^  r   z$annotations/classes-segmentation.txtrT   NrU   r   r   r   rW   rW   p   s2         K@NC@@@@@r   rW   c                   D    e Zd ZU  ed           Zeeef         ed<   dS )OpenImagesObjV5Cfgc                      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   s    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   sO         #e 5 5   FDdO     r   rY   c                   `    e Zd ZU dZeed<   dZeed<    ed           Z	e
eef         ed<   dS )	OpenImagesObjChallenge2019Cfgi  r   zEannotations/challenge-2019/challenge-2019-classes-description-500.csvrT   c                      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>   s    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      so         KaNCaaa#e 5 5   FDdO     r   re   c                   R    e Zd ZU dZeed<    ed           Zee	e
f         ed<   dS )OpenImagesSegV5Cfg,  r   c                  n    t          t                      t                      t                                S Nr4   r'   r   r   r   r(   zOpenImagesSegV5Cfg.<lambda>   *    DffFFVV= = = r   r)   r   N)r   r   r   r   r   r   r   r   r   r   r   r   r   r   ri   ri      s_         K#e 5 5   FDdO     r   ri   c                   `    e Zd ZU dZeed<   dZeed<    ed           Z	e
eef         ed<   dS )	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      so         KhM3hhh#e 5 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      sP    
			 ( ( ( ( ( ( ( (       # # # # # # # #     '        '    # # # # # # # #                          # # # # # # # # G G G G G} G G G
 A A A A A} A A A
     )   (     $4   *     )        $4     r   