|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from typing import TYPE_CHECKING |
|
|
|
from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available, is_vision_available |
|
|
|
|
|
_import_structure = { |
|
"configuration_conditional_detr": [ |
|
"CONDITIONAL_DETR_PRETRAINED_CONFIG_ARCHIVE_MAP", |
|
"ConditionalDetrConfig", |
|
"ConditionalDetrOnnxConfig", |
|
] |
|
} |
|
|
|
try: |
|
if not is_vision_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
_import_structure["feature_extraction_conditional_detr"] = ["ConditionalDetrFeatureExtractor"] |
|
_import_structure["image_processing_conditional_detr"] = ["ConditionalDetrImageProcessor"] |
|
|
|
try: |
|
if not is_torch_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
_import_structure["modeling_conditional_detr"] = [ |
|
"CONDITIONAL_DETR_PRETRAINED_MODEL_ARCHIVE_LIST", |
|
"ConditionalDetrForObjectDetection", |
|
"ConditionalDetrForSegmentation", |
|
"ConditionalDetrModel", |
|
"ConditionalDetrPreTrainedModel", |
|
] |
|
|
|
|
|
if TYPE_CHECKING: |
|
from .configuration_conditional_detr import ( |
|
CONDITIONAL_DETR_PRETRAINED_CONFIG_ARCHIVE_MAP, |
|
ConditionalDetrConfig, |
|
ConditionalDetrOnnxConfig, |
|
) |
|
|
|
try: |
|
if not is_vision_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
from .feature_extraction_conditional_detr import ConditionalDetrFeatureExtractor |
|
from .image_processing_conditional_detr import ConditionalDetrImageProcessor |
|
|
|
try: |
|
if not is_torch_available(): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
pass |
|
else: |
|
from .modeling_conditional_detr import ( |
|
CONDITIONAL_DETR_PRETRAINED_MODEL_ARCHIVE_LIST, |
|
ConditionalDetrForObjectDetection, |
|
ConditionalDetrForSegmentation, |
|
ConditionalDetrModel, |
|
ConditionalDetrPreTrainedModel, |
|
) |
|
|
|
else: |
|
import sys |
|
|
|
sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__) |
|
|