Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
msu.create_float_list_feature_list(name + "_ymax", REGION_BBOX_YMAX_KEY,
prefix=prefix, module_dict=globals())
msu.create_float_list_feature_list(name + "_xmax", REGION_BBOX_XMAX_KEY,
prefix=prefix, module_dict=globals())
msu.create_float_list_feature_list(name + "_point_x", REGION_POINT_X_KEY,
prefix=prefix, module_dict=globals())
msu.create_float_list_feature_list(name + "_point_y", REGION_POINT_Y_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_list_context_feature(name + "_parts",
REGION_PARTS_KEY,
prefix=prefix, module_dict=globals())
msu.create_float_list_context_feature(
name + "_embedding_dimensions_per_region",
REGION_EMBEDDING_DIMENSIONS_PER_REGION_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_context_feature(name + "_embedding_format",
REGION_EMBEDDING_FORMAT_KEY,
prefix=prefix, module_dict=globals())
msu.create_float_list_feature_list(name + "_embedding_floats",
REGION_EMBEDDING_FLOAT_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_list_feature_list(name + "_embedding_encoded",
REGION_EMBEDDING_ENCODED_KEY,
prefix=prefix, module_dict=globals())
msu.create_float_list_feature_list(name + "_embedding_confidence",
REGION_EMBEDDING_CONFIDENCE_KEY,
prefix=prefix, module_dict=globals())
# pylint: disable=undefined-variable
def get_prefixed_bbox_at(index, sequence_example, prefix):
return np.stack((
get_bbox_ymin_at(index, sequence_example, prefix=prefix),
get_bbox_xmin_at(index, sequence_example, prefix=prefix),
msu.create_int_context_feature(name + "_width", IMAGE_WIDTH_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_feature_list(name + "_encoded", IMAGE_ENCODED_KEY,
prefix=prefix, module_dict=globals())
msu.create_float_context_feature(name + "_frame_rate", IMAGE_FRAME_RATE_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_list_context_feature(
name + "_class_label_string", IMAGE_CLASS_LABEL_STRING_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_list_context_feature(
name + "_class_label_index", IMAGE_CLASS_LABEL_INDEX_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_list_context_feature(
name + "_object_class_index", IMAGE_OBJECT_CLASS_INDEX_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_context_feature(name + "_data_path", IMAGE_DATA_PATH_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_feature_list(name + "_timestamp", IMAGE_TIMESTAMP_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_list_feature_list(name + "_multi_encoded",
IMAGE_MULTI_ENCODED_KEY, prefix=prefix,
module_dict=globals())
FORWARD_FLOW_PREFIX = "FORWARD_FLOW"
def _create_image_with_prefix(name, prefix):
"""Create multiple accessors for image based data."""
msu.create_bytes_context_feature(name + "_format", IMAGE_FORMAT_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_context_feature(name + "_colorspace", IMAGE_COLORSPACE_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_context_feature(name + "_channels", IMAGE_CHANNELS_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_context_feature(name + "_height", IMAGE_HEIGHT_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_context_feature(name + "_width", IMAGE_WIDTH_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_feature_list(name + "_encoded", IMAGE_ENCODED_KEY,
prefix=prefix, module_dict=globals())
msu.create_float_context_feature(name + "_frame_rate", IMAGE_FRAME_RATE_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_list_context_feature(
name + "_class_label_string", IMAGE_CLASS_LABEL_STRING_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_list_context_feature(
name + "_class_label_index", IMAGE_CLASS_LABEL_INDEX_KEY,
def _create_image_with_prefix(name, prefix):
"""Create multiple accessors for image based data."""
msu.create_bytes_context_feature(name + "_format", IMAGE_FORMAT_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_context_feature(name + "_colorspace", IMAGE_COLORSPACE_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_context_feature(name + "_channels", IMAGE_CHANNELS_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_context_feature(name + "_height", IMAGE_HEIGHT_KEY,
prefix=prefix, module_dict=globals())
msu.create_int_context_feature(name + "_width", IMAGE_WIDTH_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_feature_list(name + "_encoded", IMAGE_ENCODED_KEY,
prefix=prefix, module_dict=globals())
msu.create_float_context_feature(name + "_frame_rate", IMAGE_FRAME_RATE_KEY,
prefix=prefix, module_dict=globals())
msu.create_bytes_list_context_feature(
name + "_class_label_string", IMAGE_CLASS_LABEL_STRING_KEY,
prefix=prefix, module_dict=globals())
CLIP_LABEL_STRING_KEY = "clip/label/string"
# A list of label confidences for this clip.
CLIP_LABEL_CONFIDENCE_KEY = "clip/label/confidence"
msu.create_bytes_context_feature(
"example_id", EXAMPLE_ID_KEY, module_dict=globals())
msu.create_bytes_context_feature(
"example_dataset_name", EXAMPLE_DATASET_NAME_KEY, module_dict=globals())
msu.create_bytes_context_feature(
"clip_media_id", CLIP_MEDIA_ID_KEY, module_dict=globals())
msu.create_bytes_context_feature(
"clip_alternative_media_id", ALTERNATIVE_CLIP_MEDIA_ID_KEY,
module_dict=globals())
msu.create_bytes_context_feature(
"clip_encoded_media_bytes", CLIP_ENCODED_MEDIA_BYTES_KEY,
module_dict=globals())
msu.create_bytes_context_feature(
"clip_data_path", CLIP_DATA_PATH_KEY, module_dict=globals())
msu.create_int_context_feature(
"clip_encoded_media_start_timestamp",
CLIP_ENCODED_MEDIA_START_TIMESTAMP_KEY, module_dict=globals())
msu.create_int_context_feature(
"clip_start_timestamp", CLIP_START_TIMESTAMP_KEY, module_dict=globals())
msu.create_int_context_feature(
"clip_end_timestamp", CLIP_END_TIMESTAMP_KEY, module_dict=globals())
msu.create_bytes_list_context_feature(
"clip_label_string", CLIP_LABEL_STRING_KEY, module_dict=globals())
msu.create_int_list_context_feature(
"clip_label_index", CLIP_LABEL_INDEX_KEY, module_dict=globals())
msu.create_float_list_context_feature(
"clip_label_confidence", CLIP_LABEL_CONFIDENCE_KEY, module_dict=globals())
################################## SEGMENTS #################################
CLIP_ENCODED_MEDIA_START_TIMESTAMP_KEY = "clip/encoded_media_start_timestamp"
# The start time, in microseconds, for the start of the clip in the media.
CLIP_START_TIMESTAMP_KEY = "clip/start/timestamp"
# The end time, in microseconds, for the end of the clip in the media.
CLIP_END_TIMESTAMP_KEY = "clip/end/timestamp"
# A list of label indices for this clip.
CLIP_LABEL_INDEX_KEY = "clip/label/index"
# A list of label strings for this clip.
CLIP_LABEL_STRING_KEY = "clip/label/string"
# A list of label confidences for this clip.
CLIP_LABEL_CONFIDENCE_KEY = "clip/label/confidence"
msu.create_bytes_context_feature(
"example_id", EXAMPLE_ID_KEY, module_dict=globals())
msu.create_bytes_context_feature(
"example_dataset_name", EXAMPLE_DATASET_NAME_KEY, module_dict=globals())
msu.create_bytes_context_feature(
"clip_media_id", CLIP_MEDIA_ID_KEY, module_dict=globals())
msu.create_bytes_context_feature(
"clip_alternative_media_id", ALTERNATIVE_CLIP_MEDIA_ID_KEY,
module_dict=globals())
msu.create_bytes_context_feature(
"clip_encoded_media_bytes", CLIP_ENCODED_MEDIA_BYTES_KEY,
module_dict=globals())
msu.create_bytes_context_feature(
"clip_data_path", CLIP_DATA_PATH_KEY, module_dict=globals())
msu.create_int_context_feature(
"clip_encoded_media_start_timestamp",
CLIP_ENCODED_MEDIA_START_TIMESTAMP_KEY, module_dict=globals())
msu.create_int_context_feature(
"clip_start_timestamp", CLIP_START_TIMESTAMP_KEY, module_dict=globals())
msu.create_int_context_feature(
"clip_end_timestamp", CLIP_END_TIMESTAMP_KEY, module_dict=globals())
# The feature as a list of ints.
FEATURE_INTS_KEY = "feature/ints"
# The timestamp, in microseconds, of the feature.
FEATURE_TIMESTAMP_KEY = "feature/timestamp"
# It is occasionally useful to indicate that a feature applies to a given range.
# This should be used for features only and annotations should be provided as
# segments.
FEATURE_DURATION_KEY = "feature/duration"
# Encodes an optional confidence score for the generated features.
FEATURE_CONFIDENCE_KEY = "feature/confidence"
msu.create_int_list_context_feature(
"feature_dimensions", FEATURE_DIMENSIONS_KEY, module_dict=globals())
msu.create_float_context_feature(
"feature_rate", FEATURE_RATE_KEY, module_dict=globals())
msu.create_bytes_context_feature(
"feature_bytes_format", FEATURE_BYTES_FORMAT_KEY, module_dict=globals())
msu.create_float_context_feature(
"feature_sample_rate", FEATURE_SAMPLE_RATE_KEY, module_dict=globals())
msu.create_int_context_feature(
"feature_num_channels", FEATURE_NUM_CHANNELS_KEY, module_dict=globals())
msu.create_int_context_feature(
"feature_num_samples", FEATURE_NUM_SAMPLES_KEY, module_dict=globals())
msu.create_float_context_feature(
"feature_packet_rate", FEATURE_PACKET_RATE_KEY, module_dict=globals())
msu.create_float_context_feature(
"feature_audio_sample_rate", FEATURE_AUDIO_SAMPLE_RATE_KEY,
module_dict=globals())
msu.create_float_list_feature_list(
"feature_floats", FEATURE_FLOATS_KEY, module_dict=globals())
msu.create_bytes_list_feature_list(
"feature_bytes", FEATURE_BYTES_KEY, module_dict=globals())