mirror of
https://github.com/alibaba/EasyCV.git
synced 2025-06-03 14:49:00 +08:00
b198c5a81f
dino_4sc_r50_12e：48.71 dino_4sc_r50_24e：50.53 dino_4sc_r50_36e：50.69 dino_4sc_swinl_12e: 56.86 dino_4sc_swinl_36e: 58.04 dino_5sc_swinl_36e: 58.47
25 lines
1019 B
Python
25 lines
1019 B
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
import logging
|
|
|
|
from easycv.models.detection.detectors.dab_detr import (DABDETRHead,
|
|
DABDetrTransformer)
|
|
from easycv.models.detection.detectors.detection import Detection
|
|
from easycv.models.detection.detectors.detr import DETRHead, DetrTransformer
|
|
from easycv.models.detection.detectors.dino import (DeformableTransformer,
|
|
DINOHead)
|
|
from easycv.models.detection.detectors.fcos import FCOSHead
|
|
|
|
try:
|
|
from easycv.models.detection.detectors.yolox.yolox import YOLOX
|
|
except Exception as e:
|
|
logging.info(f'Exception: {e}')
|
|
logging.info(
|
|
'Import YOLOX failed! please check your CUDA & Pytorch Version')
|
|
|
|
try:
|
|
from easycv.models.detection.detectors.yolox_edge.yolox_edge import YOLOX_EDGE
|
|
except Exception as e:
|
|
logging.info(f'Exception: {e}')
|
|
logging.info(
|
|
'Import YOLOX_EDGE failed! please check your CUDA & Pytorch Version')
|