first commit

rtdetrv2_pytorch/src/solver/__init__.py

@@ -0,0 +1,15 @@
+"""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,
+}