first commit
This commit is contained in:
commit
54ccfb11fe
218 files changed
+19525
No files matched your search
Executable
+10
@@ -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)
|
||||
Reference in new issue
Block a user