Skip to content

Commit

Permalink
Refactor D-FINE module structure by removing obsolete detector file a…
Browse files Browse the repository at this point in the history
…nd reorganizing imports
  • Loading branch information
eugene123tw committed Dec 19, 2024
1 parent 53c635d commit fe1e7f9
Show file tree
Hide file tree
Showing 10 changed files with 257 additions and 261 deletions.
File renamed without changes.
6 changes: 3 additions & 3 deletions src/otx/algo/detection/d_fine.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,11 +14,11 @@
from torchvision.ops import box_convert
from torchvision.tv_tensors import BoundingBoxFormat

from otx.algo.detection.backbones.hgnetv2 import HGNetv2
from otx.algo.detection.detectors import DETR
from otx.algo.detection.detectors.dfine.dfine_criterion import DFINECriterion
from otx.algo.detection.detectors.dfine.dfine_decoder import DFINETransformer
from otx.algo.detection.detectors.dfine.hgnetv2 import HGNetv2
from otx.algo.detection.detectors.dfine.hybrid_encoder import HybridEncoder
from otx.algo.detection.losses.dfine_loss import DFINECriterion
from otx.algo.detection.necks.dfine_hybrid_encoder import HybridEncoder
from otx.algo.utils.mmengine_utils import load_checkpoint
from otx.core.config.data import TileConfig
from otx.core.data.entity.base import OTXBatchLossEntity
Expand Down
4 changes: 0 additions & 4 deletions src/otx/algo/detection/detectors/dfine/__init__.py

This file was deleted.

242 changes: 0 additions & 242 deletions src/otx/algo/detection/detectors/dfine/utils.py

This file was deleted.

11 changes: 10 additions & 1 deletion src/otx/algo/detection/heads/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,19 @@
"""Custom head implementations for detection task."""

from .atss_head import ATSSHead
from .dfine_decoder import DFINETransformer
from .rtdetr_decoder import RTDETRTransformer
from .rtmdet_head import RTMDetSepBNHead
from .ssd_head import SSDHead
from .yolo_head import YOLOHead
from .yolox_head import YOLOXHead

__all__ = ["ATSSHead", "RTDETRTransformer", "RTMDetSepBNHead", "SSDHead", "YOLOHead", "YOLOXHead"]
__all__ = [
"ATSSHead",
"DFINETransformer",
"RTDETRTransformer",
"RTMDetSepBNHead",
"SSDHead",
"YOLOHead",
"YOLOXHead",
]
Loading

0 comments on commit fe1e7f9

Please sign in to comment.