File size: 388 Bytes
3de7bf6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
"""Feature extractors."""
# Copyright (C) 2022-2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
from .timm import TimmFeatureExtractor
from .torchfx import BackboneParams, TorchFXFeatureExtractor
from .utils import dryrun_find_featuremap_dims
__all__ = [
"BackboneParams",
"dryrun_find_featuremap_dims",
"TimmFeatureExtractor",
"TorchFXFeatureExtractor",
]
|