prepare new detection model

zyddnys · Jan 20, 2024 · f622f24 · f622f24
1 parent 0d43cf4
commit f622f24
Show file tree

Hide file tree

Showing 2 changed files with 598 additions and 0 deletions.
diff --git a/manga_translator/detection/__init__.py b/manga_translator/detection/__init__.py
@@ -1,13 +1,15 @@
 import numpy as np
 
 from .default import DefaultDetector
+from .dbnet_convnext import DBConvNextDetector
 from .ctd import ComicTextDetector
 from .craft import CRAFTDetector
 from .none import NoneDetector
 from .common import CommonDetector, OfflineDetector
 
 DETECTORS = {
     'default': DefaultDetector,
+    'dbconvnext': DBConvNextDetector,
     'ctd': ComicTextDetector,
     'craft': CRAFTDetector,
     'none': NoneDetector,