detection: Make model dir before downloading

2024-09-01 22:31:34 +01:00 · 2024-09-01 22:31:34 +01:00 · 45432648c2
commit 45432648c2
parent cc33eebbe5
1 changed files with 1 additions and 0 deletions
--- a/detection/utils.py
+++ b/detection/utils.py
@ -2,6 +2,7 @@ import os
 import oss2

 def get_model(name):
+    os.makedirs(os.path.join(os.path.dirname(__file__), "model"), exist_ok=True)
    fn = os.path.join(os.path.dirname(__file__), "model", name)
    if os.path.exists(fn):
        return fn