def mask_valid_voxels(inputs, outputs): """Mask the voxels that are valid.""" if standard_fields.DetectionResultFields.objects_center in outputs: return outputs valid_mask = mask_utils.num_voxels_mask(inputs=inputs) mask_utils.apply_mask_to_output_voxel_tensors(outputs=outputs, valid_mask=valid_mask) for key, value in standard_fields.get_output_voxel_to_object_field_mapping( ).items(): if key in outputs: outputs[value] = outputs[key]
def _get_voxels_valid_mask(inputs_1): """Returns the mask that removes voxels that are not within image.""" return mask_utils.num_voxels_mask(inputs=inputs_1)
def _get_voxels_valid_mask(inputs_1): """Returns the mask that removes voxels that are outside objects.""" num_voxels_mask = mask_utils.num_voxels_mask(inputs=inputs_1) within_objects_mask = mask_utils.voxels_within_objects_mask(inputs=inputs_1) return tf.logical_and(within_objects_mask, num_voxels_mask)
def mask_valid_voxels(inputs, outputs): """Mask the voxels that are valid and in image view.""" valid_mask = mask_utils.num_voxels_mask(inputs=inputs) mask_utils.apply_mask_to_output_voxel_tensors(outputs=outputs, valid_mask=valid_mask)