first commit

This commit is contained in:
Semi212 committed 2021-09-15 11:33:58 +08:00
commit 54ccfb11fe
218 files changed
+19525

No files matched your search

+10
View File
@@ -0,0 +1,10 @@
from .ssd_detector import SSDDetector
_DETECTION_META_ARCHITECTURES = {
"SSDDetector": SSDDetector
}
def build_detection_model(cfg):
meta_arch = _DETECTION_META_ARCHITECTURES[cfg.MODEL.META_ARCHITECTURE]
return meta_arch(cfg)