def _run_upsample(msg, run_upload=True):
"""Runs disparity upsampling according to parameters read from the message.
Args:
msg (dict[str, str]): Message received from RabbitMQ publisher.
run_upload (bool, optional): Whether or not an upload was performed.
Returns:
tuple(bool, bool): Respectively whether or not a download and upload were performed.
"""
image_types_to_level = [(msg["image_type"], msg["level"])]
msg_cp = copy(msg)
if msg["image_type"] == "disparity":
color_image_type = "color"
image_types_to_level += [
("foreground_masks", msg["level"]),
("foreground_masks", msg["dst_level"]),
]
msg_cp["foreground_masks_in"] = local_image_type_path(
msg, "foreground_masks", msg["level"])
msg_cp["foreground_masks_out"] = local_image_type_path(
msg, "foreground_masks", msg["dst_level"])
msg_cp["background_disp"] = local_image_type_path(
msg, "background_disp", msg["dst_level"])
download_image_type(msg, "background_disp", [msg["background_frame"]],
msg["dst_level"])
msg_cp["background_disp"] = local_image_type_path(
msg, "background_disp", msg["dst_level"])
elif msg["image_type"] == "background_disp":
color_image_type = "background_color"
msg_cp[
"foreground_masks_in"] = "" # Background upsampling doesn't use masks
msg_cp["foreground_masks_out"] = ""
image_types_to_level.append((color_image_type, msg["dst_level"]))
ran_download = download_image_types(msg, image_types_to_level)
ran_download |= download_rig(msg)
msg_cp["disparity"] = local_image_type_path(msg, msg["image_type"],
msg["level"])
msg_cp["output"] = local_image_type_path(
msg, config.type_to_upsample_type[msg["image_type"]])
msg_cp["color"] = local_image_type_path(msg, color_image_type,
msg["dst_level"])
_run_bin(msg_cp)
if run_upload:
ran_upload = upload_image_type(
msg, config.type_to_upsample_type[msg["image_type"]])
return ran_download, ran_upload
def upsample_layer_disparity_callback(msg):
"""Runs disparity upsampling and layering according to parameters read from the message.
Args:
msg (dict[str, str]): Message received from RabbitMQ publisher.
"""
print("Running disparity upsampling and layering...")
msg_cp = copy(msg)
msg_cp["app"] = "UpsampleDisparity"
ran_download, _ = _run_upsample(msg, run_upload=False)
ran_download |= download_image_type(
msg, config.type_to_upsample_type["background_disp"],
[msg["background_frame"]])
msg_cp["app"] = "LayerDisparities"
msg_cp["background_disp"] = local_image_type_path(
msg, config.type_to_upsample_type["background_disp"])
msg_cp["foreground_disp"] = local_image_type_path(
msg, config.type_to_upsample_type["disparity"])
msg_cp["output"] = config.DOCKER_OUTPUT_ROOT
_run_bin(msg_cp)
ran_upload = upload_image_type(msg, "disparity")
_clean_worker(ran_download, ran_upload)
def generate_foreground_masks_callback(msg):
"""Runs foreground mask generation according to parameters read from the message.
Args:
msg (dict[str, str]): Message received from RabbitMQ publisher.
"""
print("Running foreground mask generation...")
image_types_to_level = [("color", msg["level"])]
ran_download = download_rig(msg)
ran_download |= download_image_types(msg, image_types_to_level)
ran_download |= download_image_type(msg, "background_color",
[msg["background_frame"]],
msg["level"])
msg_cp = copy(msg)
msg_cp["color"] = local_image_type_path(msg, "color", msg["level"])
msg_cp["background_color"] = local_image_type_path(msg, "background_color",
msg["level"])
msg_cp["foreground_masks"] = local_image_type_path(msg, "foreground_masks",
msg["dst_level"])
_run_bin(msg_cp)
ran_upload = upload_image_type(msg,
"foreground_masks",
level=msg["dst_level"])
_clean_worker(ran_download, ran_upload)
def simple_mesh_renderer_callback(msg):
print("Generating exports...")
msg_cp = copy(msg)
frames = get_frame_range(msg_cp["first"], msg_cp["last"])
ran_download = download_rig(msg)
ran_download = download_image_type(msg, msg_cp["color_type"], frames)
ran_download |= download_image_type(msg, msg_cp["disparity_type"], frames)
msg_cp["color"] = local_image_type_path(msg, msg_cp["color_type"])
msg_cp["disparity"] = local_image_type_path(msg, msg_cp["disparity_type"])
msg_cp["output"] = local_image_type_path(msg, msg_cp["dst_image_type"])
msg_cp["position"] = '"0.0 0.0 0.0"'
msg_cp["forward"] = '"-1.0 0.0 0.0"'
msg_cp["up"] = '"0.0 0.0 1.0"'
_run_bin(msg_cp)
ran_upload = upload_image_type(msg, msg_cp["dst_image_type"], frames)
_clean_worker(ran_download, ran_upload)
def depth_estimation_callback(msg):
"""Runs depth estimation according to parameters read from the message.
Args:
msg (dict[str, str]): Message received from RabbitMQ publisher.
"""
print("Running depth estimation...")
ran_download = False
msg_cp = copy(msg)
if msg["image_type"] == "disparity":
image_types_to_level = [("color", msg["level_start"])]
if msg["use_foreground_masks"]:
ran_download |= download_image_type(msg, "background_disp",
[msg["background_frame"]],
msg["level_start"])
image_types_to_level.append(
("foreground_masks", msg["level_start"]))
if msg["level_start"] < msg["num_levels"] - 1:
image_types_to_level.append(("disparity", msg["level_start"] + 1))
if msg["use_foreground_masks"]:
image_types_to_level.append(
("foreground_masks", msg["level_start"] + 1))
else:
image_types_to_level = [("background_color", msg["level_start"])]
if msg["level_start"] < msg["num_levels"] - 1:
image_types_to_level.append(
("background_disp", msg["level_start"] + 1))
msg_cp["color"] = local_image_type_path(msg, "background_color_levels")
msg_cp["output_root"] = os.path.join(msg["input_root"], "background")
ran_download |= download_rig(msg)
ran_download |= download_image_types(msg, image_types_to_level)
_run_bin(msg_cp)
ran_upload = upload_image_type(msg,
msg["image_type"],
level=msg["level_end"])
_clean_worker(ran_download, ran_upload)
