|
|
@@ -1,10 +1,15 @@
|
|
|
+import logging
|
|
|
+import numpy as np
|
|
|
import os
|
|
|
import warnings
|
|
|
+import yaml
|
|
|
|
|
|
from cvargparse import BaseParser
|
|
|
from cvdatasets.utils import read_info_file
|
|
|
from functools import wraps
|
|
|
|
|
|
+from pathlib import Path
|
|
|
+
|
|
|
|
|
|
WARNING = """Could not find default info file \"{}\". """ + \
|
|
|
"""Some arguments (dataset, parts etc.) are not restraint to certain choices! """ + \
|
|
|
@@ -32,3 +37,52 @@ def parser_extender(extender):
|
|
|
return parser
|
|
|
|
|
|
return inner
|
|
|
+
|
|
|
+
|
|
|
+def populate_args(args, ignore = None, replace = {}, fc_params = []):
|
|
|
+
|
|
|
+ args.debug = False
|
|
|
+
|
|
|
+ assert args.load is not None, "--load argument missing!"
|
|
|
+
|
|
|
+ model_path = Path(args.load)
|
|
|
+
|
|
|
+ args_path = model_path.parent / "meta" / "args.yml"
|
|
|
+
|
|
|
+ assert args_path.exists(), f"Couldn't find args file \"{args_path}\""
|
|
|
+
|
|
|
+ logging.info(f"Reading arguments from \"{args_path}\"")
|
|
|
+
|
|
|
+ with open(args_path) as f:
|
|
|
+ dumped_args = yaml.safe_load(f)
|
|
|
+
|
|
|
+ for key, value in dumped_args.items():
|
|
|
+ if (ignore is not None and key in ignore) or getattr(args, key, None) == value:
|
|
|
+ continue
|
|
|
+
|
|
|
+ if key in ["logfile", "loglevel"]:
|
|
|
+ continue
|
|
|
+
|
|
|
+ old_value = getattr(args, key, None)
|
|
|
+ if key in replace:
|
|
|
+ value = replace[key].get(value, value)
|
|
|
+
|
|
|
+ logging.debug(f"Setting \"{key}\" to {value} (originally was {'missing' if old_value is None else old_value})")
|
|
|
+
|
|
|
+ setattr(args, key, value)
|
|
|
+
|
|
|
+ # get the correct number of classes
|
|
|
+ args.n_classes = 1000
|
|
|
+ weights = np.load(args.load)
|
|
|
+ n_classes_found = False
|
|
|
+ for key in fc_params:
|
|
|
+ try:
|
|
|
+ args.n_classes = weights[key].shape[0]
|
|
|
+ n_classes_found = True
|
|
|
+ break
|
|
|
+ except KeyError as e:
|
|
|
+ pass
|
|
|
+
|
|
|
+ if not n_classes_found:
|
|
|
+ raise KeyError("Could not find number of classes!")
|
|
|
+
|
