Python num_voxels_mask Examples

Example #1

File: model_utils.py Project: ziyouzizai111/google-research

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]

Example #2

File: classification_losses.py Project: ziyouzizai111/google-research

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)

Example #3

File: box_prediction_losses.py Project: ziyouzizai111/google-research

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)

Example #4

File: model_utils.py Project: zrqohbug/google-research

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)