^j-'8dZddlmZgdZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZddlmZeZedd d Gd d eZGddeZGddeZGddeZ GddeZ!edd d GddeZ"GddeZ#GddeZ$edd d Gdde$Z%Gdde$Z&Gd d!e$Z'Gd"d#e$Z(Gd$d%e$Z)Gd&d'e$Z*Gd(d)e$Z+y)*aVConcrete RF-DETR model variant classes. All classes inherit from :class:`~rfdetr.detr.RFDETR` which remains defined in ``rfdetr.detr``. Backward-compatible access from ``rfdetr.detr`` is provided via lazy ``__getattr__`` re-exports, so importing ``rfdetr.variants`` no longer depends on a fragile eager ``detr -> variants`` import sequence. ) annotations) RFDETRBaseRFDETRKeypointPreview RFDETRNano RFDETRSmall RFDETRMedium RFDETRLargeRFDETRLargeDeprecated RFDETRSegRFDETRSegPreview RFDETRSegNanoRFDETRSegSmallRFDETRSegMediumRFDETRSegLargeRFDETRSegXLargeRFDETRSeg2XLarge)deprecated_class)KeypointTrainConfig ModelConfigRFDETRBaseConfigRFDETRKeypointPreviewConfigRFDETRLargeConfigRFDETRLargeDeprecatedConfigRFDETRMediumConfigRFDETRNanoConfigRFDETRSeg2XLargeConfigRFDETRSegLargeConfigRFDETRSegMediumConfigRFDETRSegNanoConfigRFDETRSegPreviewConfigRFDETRSegSmallConfigRFDETRSegXLargeConfigRFDETRSmallConfigSegmentationTrainConfig)RFDETR) get_loggerNz1.7.0z2.0.0)target deprecated_in remove_inceZdZdZdZeZy)rzTrain an RF-DETR Base model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``. z rfdetr-baseN)__name__ __module__ __qualname____doc__sizer_model_config_classZ/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/rfdetr/variants.pyrr<s  D*r2rceZdZdZdZeZy)rzTrain an RF-DETR Nano model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``. z rfdetr-nanoN)r+r,r-r.r/rr0r1r2r3rrLs D*r2rceZdZdZdZeZeZy)rzTrain or run inference with the RF-DETR keypoint preview model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``. zrfdetr-keypoint-previewN) r+r,r-r.r/rr0r_train_config_classr1r2r3rrWs %D5-r2rceZdZdZdZeZy)rzTrain an RF-DETR Small model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``. z rfdetr-smallN)r+r,r-r.r/r#r0r1r2r3rrcs D+r2rceZdZdZdZeZy)rzTrain an RF-DETR Medium model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``. z rfdetr-mediumN)r+r,r-r.r/rr0r1r2r3rrns D,r2rceZdZdZdZeZy)r zTrain an RF-DETR Large model using the legacy config. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``. rfdetr-largeN)r+r,r-r.r/rr0r1r2r3r r ys  D5r2r c@eZdZdZdZeddZfdZddZxZ S)r zTrain an RF-DETR Large model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``. r:ct|jdvryt|trdvSt|trd}t fd|DSy)aReturn whether initialization should retry with deprecated Large config. The fallback is only for known checkpoint/config incompatibilities from deprecated Large weights. Runtime issues such as CUDA OOM must fail fast and must not trigger a second initialization attempt. Args: exc: Exception raised by initial ``RFDETR`` initialization. Returns: ``True`` when retrying with deprecated config is expected to help. z out of memoryF patch_size)zerror(s) in loading state_dictz size mismatchzmissing key(s) in state_dictzunexpected key(s) in state_dictc3&K|]}|v yw)Nr1).0markermessages r3 zDRFDETRLarge._should_fallback_to_deprecated_config..sWVv(Ws)strlower isinstance ValueError RuntimeErrorany)excincompatible_state_dict_markersrAs @r3%_should_fallback_to_deprecated_configz1RFDETRLarge._should_fallback_to_deprecated_configs^c(.." g % c: &7* * c< (/ + W7VWW Wr2c  d|_d|_tjdj}d|vxr|j d|k7} t |d i|y#ttf$r}|s|j|s||_d|_ t |d i|tjdddzzdzn@#t$r4}tjd|j| |j|d}~wwxYwYd}~yd}~wwxYw) NF resolutionTz = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = =z WARNING: Automatically switched to deprecated model configuration, due to using deprecated weights. This will be removed in v1.9.0. Please retrain your model with the new weights and configuration. =d zRetry with deprecated RF-DETR Large configuration failed; re-raising the original initialization error for compatibility. Original error: %s)exc_infor1) init_error is_deprecatedr model_fieldsdefaultgetsuper__init__rFrGrKloggerwarning Exception exception)selfkwargs_default_resolution_custom_resolutionrI retry_exc __class__s r3rWzRFDETRLarge.__init__s$"0<<\JRR)V3g <8PTg8g 5 G  &v &L) 5!)S)STW)X!DO!%D  5 *6*!  ! !% % 5  )OO& !oo94 5 5s<AD'"C= -B87C=8 C5/C00C55C==Dc F|js tdi|Stdi|S)Nr1)rRrr)r\r]s r3get_model_configzRFDETRLarge.get_model_configs'!!$.v. ..88 8r2)rIrZreturnbool)rdr) r+r,r-r.r/ staticmethodrKrWrc __classcell__)ras@r3r r s, D6%5N9r2r ceZdZdZeZy)r zBase class for all RF-DETR segmentation models. Training accepts custom square integer ``resolution`` values. Most segmentation variants use multiples of 24; ``RFDETRSegNano`` uses multiples of 12. N)r+r,r-r.r$r6r1r2r3r r s 2r2r ceZdZdZdZeZy)r zTrain an RF-DETR Segmentation Preview model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``. Deprecated in v1.7.0, scheduled for removal in v2.0.0. zrfdetr-seg-previewN)r+r,r-r.r/r r0r1r2r3r r s  D0r2r ceZdZdZdZeZy)r zTrain an RF-DETR Segmentation Nano model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``; this variant uses multiples of 12. zrfdetr-seg-nanoN)r+r,r-r.r/rr0r1r2r3r r s D-r2r ceZdZdZdZeZy)rzTrain an RF-DETR Segmentation Small model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``; this variant uses multiples of 24. zrfdetr-seg-smallN)r+r,r-r.r/r!r0r1r2r3rr D.r2rceZdZdZdZeZy)rzTrain an RF-DETR Segmentation Medium model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``; this variant uses multiples of 24. zrfdetr-seg-mediumN)r+r,r-r.r/rr0r1r2r3rr  D/r2rceZdZdZdZeZy)rzTrain an RF-DETR Segmentation Large model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``; this variant uses multiples of 24. zrfdetr-seg-largeN)r+r,r-r.r/rr0r1r2r3rrrlr2rceZdZdZdZeZy)rzTrain an RF-DETR Segmentation XLarge model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``; this variant uses multiples of 24. zrfdetr-seg-xlargeN)r+r,r-r.r/r"r0r1r2r3rr"rnr2rceZdZdZdZeZy)rzTrain an RF-DETR Segmentation 2XLarge model. Training accepts custom square integer ``resolution`` values. The value must be divisible by ``patch_size * num_windows``; this variant uses multiples of 24. zrfdetr-seg-2xlargeN)r+r,r-r.r/rr0r1r2r3rr-s D0r2r),r. __future__r__all__ deprecater rfdetr.configrrrrrrrrrrrrr r!r"r#r$ rfdetr.detrr%rfdetr.utilities.loggerr&rXrrrrrr r r r r rrrrrr1r2r3rxsQ # $'&.   ++  +++ .F .,&,-6-  6F6  6P9&P9f22  1y1  1.I./Y/0i0/Y/0i01y1r2