def run(self, args):
data = dataset.DataSet(args.dataset)
data.config['interactive'] = args.interactive
reconstructions = data.load_reconstruction()
graph = data.load_tracks_graph()
dense.compute_depthmaps(data, graph, reconstructions[0])
def run(data):
#udata = dataset.UndistortedDataSet(data, args.subfolder)
udata = data.undistorted_data
data.config['interactive'] = False # FAlse
reconstructions = data.udata_reconstruction
graph = data.udata_track_graph
dense.compute_depthmaps(data, udata, graph, reconstructions[0])
def run(self, args):
data = dataset.DataSet(args.dataset)
data.config['interactive'] = args.interactive
reconstructions = data.load_undistorted_reconstruction()
graph = data.load_undistorted_tracks_graph()
dense.compute_depthmaps(data, graph, reconstructions[0])
def run(self, args):
data = dataset.DataSet(args.dataset)
udata = dataset.UndistortedDataSet(data, args.subfolder)
data.config['interactive'] = args.interactive
reconstructions = udata.load_undistorted_reconstruction()
tracks_manager = udata.load_undistorted_tracks_manager()
dense.compute_depthmaps(udata, tracks_manager, reconstructions[0])
def run(self, args):
data = dataset.DataSet(args)
udata = dataset.UndistortedDataSet(data, 'undistorted')
data.config['interactive'] = args
reconstructions = udata.load_undistorted_reconstruction()
graph = udata.load_undistorted_tracks_graph()
dense.compute_depthmaps(udata, graph, reconstructions[0])
def run(self, file_path, opensfm_config):
# data = dataset.DataSet(args.dataset)
udata = opensfm_interface.UndistortedDataSet(file_path, opensfm_config,
'undistorted')
reconstructions = udata.load_undistorted_reconstruction()
graph = udata.load_undistorted_tracks_graph()
#opensfm_config['interactive'] =
dense.compute_depthmaps(udata, graph, reconstructions[0])
def _(self):
with group_stubs(
attach_stub(dense,
"lookup_neighboring_images").call_real(),
attach_stub(dense, "config_depth_range").call_real(),
attach_stub(dense, "compute_depthmap"),
attach_stub(dense, "merge_depthmaps"),
attach_stub(dense, "clean_depthmap"),
attach_stub(dense, "prune_depthmap")):
compute_depthmaps(self.data, None, self.reconstruction,
self.neighbors_dict)
expect(dense.lookup_neighboring_images).was_called()
expect(dense.config_depth_range).was_called()
def run_dataset(data, subfolder, interactive):
""" Compute depthmap on a dataset with has SfM ran already.
Args:
subfolder: dataset's subfolder where to store results
interactive : display plot of computed depthmaps
"""
udataset = dataset.UndistortedDataSet(data, subfolder)
data.config['interactive'] = interactive
reconstructions = udataset.load_undistorted_reconstruction()
tracks_manager = udataset.load_undistorted_tracks_manager()
dense.compute_depthmaps(udataset, tracks_manager, reconstructions[0])
def run_dataset(data: DataSet, subfolder, interactive) -> None:
"""Compute depthmap on a dataset with has SfM ran already.
Args:
subfolder: dataset's subfolder where to store results
interactive : display plot of computed depthmaps
"""
udata_path = os.path.join(data.data_path, subfolder)
udataset = dataset.UndistortedDataSet(data, udata_path, io_handler=data.io_handler)
data.config["interactive"] = interactive
reconstructions = udataset.load_undistorted_reconstruction()
tracks_manager = udataset.load_undistorted_tracks_manager()
dense.compute_depthmaps(udataset, tracks_manager, reconstructions[0])
def run(self, args):
data = dataset.DataSet(args.dataset)
udata = dataset.UndistortedDataSet(data, args.subfolder)
data.config['interactive'] = args.interactive
graph, neighbors_dict = None, None
reconstructions = udata.load_undistorted_reconstruction()
neighbors_path: Path = Path(data.data_path) / "neighbors.json"
if neighbors_path.exists():
with io.open_rt(neighbors_path) as fp:
neighbors_dict = io.json_load(fp)
else:
graph = udata.load_undistorted_tracks_graph()
dense.compute_depthmaps(udata, graph, reconstructions[0],
neighbors_dict)