
    ^j                         d Z ddlmZ ddlmZ ddlmZ ddlmZm	Z	m
Z
 ddlmZ de_         e       Zded	ed
edefdZded	ed
edefdZy)zDataset file for Object365.    )Path)Any)Image)CocoDetectionmake_coco_transforms"make_coco_transforms_square_div_64)
get_loggerN	image_setargs
resolutionreturnc                    t        t        |dd       xs |j                        }||dz  f||dz  fd}||    \  }}ddlm} t        |dd      }t        |d	d
      }	 ||	      }
|
d
k7  rt
        j                  d       |
d
k7  }|r2t        ||t        | ||j                  |j                  |            }|S t        ||t        | ||j                  |j                  |            }|S )Ndataset_dirz"zhiyuan_objv2_train_val_wo_5k.jsonzzhiyuan_objv2_minival5k.json)trainvalr   )resolve_augmentation_backendsquare_resize_div_64Faugmentation_backendcpuzO365 dataset does not support custom aug_config in Phase 1 GPU augmentation; Albumentations augmentation is skipped and normalization runs on GPU. Pass augmentation_backend='cpu' for full CPU augmentation pipeline with O365.)multi_scaleexpanded_scalesgpu_postprocess)
transforms)r   getattr	coco_path!rfdetr.datasets.kornia_transformsr   loggerwarningr   r   r   r   r   )r
   r   r   rootPATHS
img_folderann_filer   r   r   resolved_backendr   datasets                _/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/rfdetr/datasets/o365.pybuild_o365_rawr&      s    mT2DdnnEDCCDd;;<E !+JN"4)?G"4)?G34HI5 \	

 '%/O9 ,, $ 4 4 /

. N  + ,, $ 4 4 /

 N    c                     | dk(  rt        d||      }|S | dk(  rt        d||      }|S t        dj                  |             )Nr   )r   r   zUnknown image_set: {})r&   
ValueErrorformat)r
   r   r   train_dsval_dss        r%   
build_o365r-   I   sO    G!'4JGEt
C
,33I>
??r'   )__doc__pathlibr   typingr   PILr   rfdetr.datasets.cocor   r   r   rfdetr.utilities.loggerr	   MAX_IMAGE_PIXELSr   strintr&   r-    r'   r%   <module>r8      ss    "    h h . 	.c . .# .- .b@# @S @c @m @r'   