Pass slam config as a command line argument

pull/317/head
Pau Gargallo 2016-02-04 14:24:04 +01:00
rodzic 0201ba4497
commit 005830b53c
2 zmienionych plików z 13 dodań i 10 usunięć

Wyświetl plik

@ -37,6 +37,10 @@ parser.add_argument('--video',
metavar='<string>',
help='Path to the video file to process')
parser.add_argument('--slam-config',
metavar='<string>',
help='Path to config file for orb-slam')
parser.add_argument('--force-focal',
metavar='<positive float>',
type=float,

Wyświetl plik

@ -25,6 +25,7 @@ class ODMSlamCell(ecto.Cell):
tree = self.inputs.tree
args = self.inputs.args
video = os.path.join(tree.root_path, args['video'])
slam_config = os.path.join(tree.root_path, args['slam_config'])
if not video:
log.ODM_ERROR('No video provided')
@ -35,17 +36,16 @@ class ODMSlamCell(ecto.Cell):
system.mkdir_p(tree.pmvs)
vocabulary = os.path.join(context.orb_slam2_path, 'Vocabulary/ORBvoc.txt')
config_file = os.path.join(context.orb_slam2_path, 'Examples/Monocular/TUM1.yaml')
orb_slam_cmd = os.path.join(context.odm_modules_path, 'odm_slam')
# run slam binary
system.run(' '.join([
'cd {} &&'.format(tree.opensfm),
orb_slam_cmd,
vocabulary,
config_file,
video,
]))
# system.run(' '.join([
# 'cd {} &&'.format(tree.opensfm),
# orb_slam_cmd,
# vocabulary,
# slam_config,
# video,
# ]))
system.run(' '.join([
'cd {} &&'.format(tree.opensfm),
@ -54,11 +54,10 @@ class ODMSlamCell(ecto.Cell):
os.path.join(context.odm_modules_src_path, 'odm_slam/src/orb_slam_to_opensfm.py'),
video,
os.path.join(tree.opensfm, 'KeyFrameTrajectory.txt'),
config_file,
slam_config,
]))
### check if reconstruction was exported to bundler before
if not io.file_exists(tree.opensfm_bundle_list) or rerun_cell: