def setup(args):
cfg = new_config()
cfg.merge_from_file(f'configs/{args.config_file}.yaml')
cfg.merge_from_list(args.opts)
cfg.NAME = args.config_file
if args.log_dir: # Override Output dir
cfg.OUTPUT_DIR = path.join(args.log_dir, args.config_file)
else:
cfg.OUTPUT_DIR = path.join(cfg.OUTPUT_BASE_DIR, args.config_file)
os.makedirs(cfg.OUTPUT_DIR, exist_ok=True)
if args.data_dir:
cfg.DATASETS.TRAIN = (args.data_dir, )
return cfg
"--debug",
dest='debug',
action="store_true",
help="whether to attempt to resume from the checkpoint directory",
)
args = parser.parse_known_args()[0]
MODEL_NAME = args.config_file
CHECKPOINT_NAME = args.checkpoint_name
THRESHOLD = args.threshold
BATCH_SIZE = args.batch_size
TRAIN_TYPE = 'test'
torch.manual_seed(0)
cfg = new_config()
if args.debug:
cfg.merge_from_file(
f'/home/yonk/urban_dl/urban_dl/configs/damage_detection/{MODEL_NAME}.yaml'
)
else:
cfg.merge_from_file(
f'/Midgard/home/pshi/urban_dl/configs/damage_detection/{MODEL_NAME}.yaml'
)
cfg.OUTPUT_DIR = path.join(cfg.OUTPUT_BASE_DIR, MODEL_NAME)
net = UNet(cfg)
print('========== THRESHOLD', cfg.THRESH)
# load state dict
full_model_path = os.path.join(cfg.OUTPUT_DIR, CHECKPOINT_NAME)
def load_cfg(cfg_file: Path):
cfg = new_config()
cfg.merge_from_file(str(cfg_file))
return cfg
def load_cfg(configs_dir: Path, cfg_name: str):
configs_file = configs_dir / f'{cfg_name}.yaml'
cfg = new_config()
cfg.merge_from_file(str(configs_file))
return cfg
