in scripts/interactive_demo.py [0:0]
def main():
# import os
# os.environ["SDL_VIDEODRIVER"] = "dummy"
parser = argparse.ArgumentParser()
# parser.add_argument('--sound', default=False, action='store_true')
parser.add_argument(
"--run-type",
choices=["train", "eval"],
default='eval',
help="run type of the experiment (train or eval)",
)
parser.add_argument(
"--exp-config",
type=str,
required=False,
default='ss_baselines/av_nav/config/audionav/mp3d/interactive_demo.yaml',
help="path to config yaml containing info about experiment",
)
parser.add_argument(
"opts",
default=None,
nargs=argparse.REMAINDER,
help="Modify config options from command line",
)
parser.add_argument(
"--debug",
default=False,
action='store_true',
help="Modify config options from command line",
)
parser.add_argument(
"--keys",
default='',
type=str,
help="Modify config options from command line",
)
args = parser.parse_args()
# file_handler = logging.FileHandler(log_file, mode=mode)
stdout_handler = logging.StreamHandler(sys.stdout)
level = logging.INFO if not args.debug else logging.DEBUG
logging.basicConfig(level=level, handlers=[stdout_handler],
format='%(asctime)s, %(levelname)s: %(message)s', datefmt="%Y-%m-%d %H:%M:%S")
config = get_config(
config_paths=args.exp_config,
opts=args.opts,
run_type=args.run_type)
config.defrost()
config.TASK_CONFIG.TASK.MEASUREMENTS.append("TOP_DOWN_MAP")
config.TASK_CONFIG.DATASET.SPLIT = config.EVAL.SPLIT
if args.keys == '':
config.TASK_CONFIG.SIMULATOR.RGB_SENSOR.WIDTH = config.TASK_CONFIG.SIMULATOR.RGB_SENSOR.HEIGHT = \
config.TASK_CONFIG.SIMULATOR.DEPTH_SENSOR.WIDTH = config.TASK_CONFIG.SIMULATOR.DEPTH_SENSOR.HEIGHT = 256
config.TASK_CONFIG.SIMULATOR.CONTINUOUS_VIEW_CHANGE = False
else:
config.TASK_CONFIG.TASK.TOP_DOWN_MAP.DRAW_GOAL_POSITIONS = False
config.freeze()
print(config)
dataset = make_dataset(id_dataset=config.TASK_CONFIG.DATASET.TYPE, config=config.TASK_CONFIG.DATASET)
env = AudioNavRLEnv(config=config, dataset=dataset)
if args.keys == '':
interactive_demo(config, env)
else:
keys = args.keys.split(',')
following(config, env, keys)