Merge pull request #14 from qianlian-mozi/test

update for sdino
qianlian-mozi · Dec 18, 2023 · edde216 · edde216
2 parents 3c39cc8 + 4fb0849
commit edde216
Show file tree

Hide file tree

Showing 17 changed files with 1,565 additions and 3,708 deletions.
diff --git a/20231016_173555.log b/20231016_173555.log
diff --git a/pdino_models/__init__.py b/pdino_models/__init__.py
@@ -0,0 +1,29 @@
+# Copyright (c) OpenMMLab. All rights reserved.
+from .csp_darknet import CSPDarknet
+from .cspnext import CSPNeXt
+from .darknet import Darknet
+from .detectors_resnet import DetectoRS_ResNet
+from .detectors_resnext import DetectoRS_ResNeXt
+from .efficientnet import EfficientNet
+from .hourglass import HourglassNet
+from .hrnet import HRNet
+from .mobilenet_v2 import MobileNetV2
+from .p2t_3x_fromuppvtnew import PyramidVisionTransformer, PyramidVisionTransformerV2
+from .regnet import RegNet
+from .res2net import Res2Net
+from .resnest import ResNeSt
+from .resnet import ResNet, ResNetV1d
+from .resnext import ResNeXt
+from .ssd_vgg import SSDVGG
+from .swin import SwinTransformer
+from .trident_resnet import TridentResNet
+# from .p2t_mmdet3x import PyramidPoolingTransformer
+from .p2t_3x_fromuppvtnew import p2t_tiny
+
+__all__ = [
+    'RegNet', 'ResNet', 'ResNetV1d', 'ResNeXt', 'SSDVGG', 'HRNet',
+    'MobileNetV2', 'Res2Net', 'HourglassNet', 'DetectoRS_ResNet',
+    'DetectoRS_ResNeXt', 'Darknet', 'ResNeSt', 'TridentResNet', 'CSPDarknet',
+    'SwinTransformer', 'PyramidVisionTransformer',
+    'PyramidVisionTransformerV2', 'EfficientNet', 'CSPNeXt','p2t_tiny'
+]