Files
RT-DETR/rtdetrv2_pytorch/src/solver/__init__.py
2026-06-03 12:42:47 +08:00

15 lines
284 B
Python

"""Copyright(c) 2023 lyuwenyu. All Rights Reserved.
"""
from ._solver import BaseSolver
from .clas_solver import ClasSolver
from .det_solver import DetSolver
from typing import Dict
TASKS :Dict[str, BaseSolver] = {
'classification': ClasSolver,
'detection': DetSolver,
}