mirror of https://github.com/alibaba/EasyCV.git
17 lines
467 B
Python
17 lines
467 B
Python
|
# Copyright (c) Alibaba, Inc. and its affiliates.
|
||
|
|
import logging
|
||
|
|
|
||
|
|
try:
|
||
|
|
from .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 .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')
|