Ë
    ëÓ^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ÚcpuzàO365 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   € Ü”˜˜m¨TÓ2ÒD°d·n±nÓE€Dà˜ÐCÑCÐDØ�dÐ;Ñ;Ð<ñ€Eð ! Ñ+Ñ€J�åNä" 4Ð)?ÀÓGÐÜ" 4Ð)?ÀÓGÐÙ3Ð4HÓ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ÒÜ! '¨4¸JÔGˆØˆØ�EÒÜ  t¸
ÔCˆØˆÜ
Ð,×3Ñ3°IÓ>Ó
?Ð?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'   