kopia lustrzana https://github.com/OpenDroneMap/ODM
Full pipeline rerun test
rodzic
e80b89a055
commit
7da3e19e6d
|
@ -73,6 +73,12 @@ class OSFMContext:
|
||||||
|
|
||||||
if not io.file_exists(rs_file) or rerun:
|
if not io.file_exists(rs_file) or rerun:
|
||||||
self.run('rs_correct --output reconstruction.json --output-tracks tracks.csv')
|
self.run('rs_correct --output reconstruction.json --output-tracks tracks.csv')
|
||||||
|
|
||||||
|
log.ODM_INFO("Re-running the reconstruction pipeline")
|
||||||
|
self.match_features(True)
|
||||||
|
self.create_tracks(True)
|
||||||
|
self.reconstruct(rolling_shutter_correct=False, rerun=True)
|
||||||
|
|
||||||
self.touch(rs_file)
|
self.touch(rs_file)
|
||||||
else:
|
else:
|
||||||
log.ODM_WARNING("Rolling shutter correction already applied")
|
log.ODM_WARNING("Rolling shutter correction already applied")
|
||||||
|
@ -381,7 +387,6 @@ class OSFMContext:
|
||||||
|
|
||||||
def feature_matching(self, rerun=False):
|
def feature_matching(self, rerun=False):
|
||||||
features_dir = self.path("features")
|
features_dir = self.path("features")
|
||||||
matches_dir = self.path("matches")
|
|
||||||
|
|
||||||
if not io.dir_exists(features_dir) or rerun:
|
if not io.dir_exists(features_dir) or rerun:
|
||||||
try:
|
try:
|
||||||
|
@ -399,6 +404,10 @@ class OSFMContext:
|
||||||
else:
|
else:
|
||||||
log.ODM_WARNING('Detect features already done: %s exists' % features_dir)
|
log.ODM_WARNING('Detect features already done: %s exists' % features_dir)
|
||||||
|
|
||||||
|
self.match_features(rerun)
|
||||||
|
|
||||||
|
def match_features(self, rerun=False):
|
||||||
|
matches_dir = self.path("matches")
|
||||||
if not io.dir_exists(matches_dir) or rerun:
|
if not io.dir_exists(matches_dir) or rerun:
|
||||||
self.run('match_features')
|
self.run('match_features')
|
||||||
else:
|
else:
|
||||||
|
|
|
@ -105,8 +105,9 @@ class ODMSplitStage(types.ODM_Stage):
|
||||||
log.ODM_INFO("Reconstructing %s" % sp)
|
log.ODM_INFO("Reconstructing %s" % sp)
|
||||||
local_sp_octx = OSFMContext(sp)
|
local_sp_octx = OSFMContext(sp)
|
||||||
local_sp_octx.create_tracks(self.rerun())
|
local_sp_octx.create_tracks(self.rerun())
|
||||||
local_sp_octx.reconstruct(self.rerun())
|
local_sp_octx.reconstruct(args.rolling_shutter, self.rerun())
|
||||||
else:
|
else:
|
||||||
|
# TODO: pass rolling shutter param here
|
||||||
lre = LocalRemoteExecutor(args.sm_cluster, self.rerun())
|
lre = LocalRemoteExecutor(args.sm_cluster, self.rerun())
|
||||||
lre.set_projects([os.path.abspath(os.path.join(p, "..")) for p in submodel_paths])
|
lre.set_projects([os.path.abspath(os.path.join(p, "..")) for p in submodel_paths])
|
||||||
lre.run_reconstruction()
|
lre.run_reconstruction()
|
||||||
|
|
Ładowanie…
Reference in New Issue