update
Browse files- inference.py +3 -2
- output/out.jpg +0 -0
inference.py
CHANGED
@@ -40,8 +40,9 @@ def profile(func):
|
|
40 |
|
41 |
|
42 |
def auto_load_weight(weight, version=None, map_location=None):
|
43 |
-
"""Auto load Generator version from weight."""
|
44 |
-
|
|
|
45 |
weight_name = os.path.basename(weight).lower()
|
46 |
cached_weight = os.path.join(cache_dir, weight_name)
|
47 |
|
|
|
40 |
|
41 |
|
42 |
def auto_load_weight(weight, version=None, map_location=None):
|
43 |
+
"""Auto load Generator version from weight."""
|
44 |
+
project_dir = os.path.dirname(os.path.abspath(__file__))
|
45 |
+
cache_dir = os.path.join(project_dir, ".cache")
|
46 |
weight_name = os.path.basename(weight).lower()
|
47 |
cached_weight = os.path.join(cache_dir, weight_name)
|
48 |
|
output/out.jpg
CHANGED
![]() |
![]() |