o
    ¶N i
  ã                   @   sš   d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z G dd	„ d	ejƒZG d
d„ dejƒZG dd„ dejƒZedkrKe ¡  dS dS )zTests for coco dataset module.é    )Úabsolute_import)Údivision)Úprint_function)Útesting)ÚcocoNc                	   @   sN   e Zd ZejZdgZejj	dejj
dejjdddiZdddddd	d
ddœZdS )ÚCoco2014TestÚ2014é   é   Útest2015Útrain_imagesÚtrainval_annotationsÚ
val_imagesÚtest_imagesÚtest_annotationsÚtest2015_imagesÚtest2015_annotations)r   Útrain_annotationsÚvalidation_imagesÚvalidation_annotationsr   r   r   r   N©Ú__name__Ú
__module__Ú__qualname__r   ÚCocoÚDATASET_CLASSÚBUILDER_CONFIG_NAMES_TO_TESTÚtfdsÚSplitÚTRAINÚ
VALIDATIONÚTESTÚSPLITSÚDL_EXTRACT_RESULT© r$   r$   úb/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/object_detection/coco_test.pyr      s"    ü
ør   c                   @   sF   e Zd ZejZdgZejj	dejj
dejjdiZddddddd	œZd
S )ÚCoco2017TestÚ2017r	   r
   r   r   r   r   r   )r   r   r   r   r   r   Nr   r$   r$   r$   r%   r&   1   s    ý
úr&   c                   @   s:   e Zd ZejZdgZejj	dejj
diZdddddœZdS )	ÚCoco2017PanopticTestÚ2017_panopticé   r
   r   r   Ú!panoptic_annotations_trainval2017)r   r   r   r   N)r   r   r   r   r   r   r   r   r   r   r    r"   r#   r$   r$   r$   r%   r(   C   s    þ
ür(   Ú__main__)Ú__doc__Ú
__future__r   r   r   Útensorflow_datasetsr   Ú$tensorflow_datasets.object_detectionr   Útensorflow_datasets.public_apiÚ
public_apir   ÚDatasetBuilderTestCaser   r&   r(   r   Ú	test_mainr$   r$   r$   r%   Ú<module>   s   ÿ