From a11992ab0f211a9e2442c4f191854a22d10f3e61 Mon Sep 17 00:00:00 2001 From: Esteban Date: Wed, 5 Apr 2023 11:21:57 +0200 Subject: [PATCH] mod: Downloading models to ProgramData folder under windows --- opendm/ai.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/opendm/ai.py b/opendm/ai.py index 834d4043..31c31f81 100644 --- a/opendm/ai.py +++ b/opendm/ai.py @@ -3,11 +3,16 @@ from opendm.net import download from opendm import log import zipfile import time +import sys, platform def get_model(namespace, url, version, name = "model.onnx"): version = version.replace(".", "_") - base_dir = os.path.join(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")), "storage", "models") + base_dir = os.path.join(os.path.dirname(__file__), "..") + if sys.platform == 'win32': + base_dir = os.path.join(os.getenv('PROGRAMDATA'),"ODM") + base_dir = os.path.join(os.path.abspath(base_dir), "storage", "models") + namespace_dir = os.path.join(base_dir, namespace) versioned_dir = os.path.join(namespace_dir, version)