Spaces:
Sleeping
Sleeping
File size: 4,327 Bytes
fa84113 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
class InputDataFields(object):
"""Names for the input tensors.
Holds the standard data field names to use for identifying input tensors. This
should be used by the decoder to identify keys for the returned tensor_dict
containing input tensors. And it should be used by the model to identify the
tensors it needs.
Attributes:
image: image.
image_additional_channels: additional channels.
key: unique key corresponding to image.
filename: original filename of the dataset (without common path).
gt_image_classes: image-level class labels.
gt_image_confidences: image-level class confidences.
gt_labeled_classes: image-level annotation that indicates the
classes for which an image has been labeled.
gt_boxes: coordinates of the ground truth boxes in the image.
gt_classes: box-level class labels.
gt_confidences: box-level class confidences. The shape should be
the same as the shape of gt_classes.
gt_label_types: box-level label types (e.g. explicit negative).
gt_is_crowd: [DEPRECATED, use gt_group_of instead]
is the groundtruth a single object or a crowd.
gt_area: area of a groundtruth segment.
gt_difficult: is a `difficult` object
gt_group_of: is a `group_of` objects, e.g. multiple objects of the
same class, forming a connected group, where instances are heavily
occluding each other.
gt_instance_masks: ground truth instance masks.
gt_instance_boundaries: ground truth instance boundaries.
gt_instance_classes: instance mask-level class labels.
gt_label_weights: groundtruth label weights.
gt_weights: groundtruth weight factor for bounding boxes.
image_height: height of images, used to decode
image_width: width of images, used to decode
"""
image = 'image'
key = 'image_id'
filename = 'filename'
gt_boxes = 'bbox'
gt_classes = 'cls'
gt_confidences = 'confidences'
gt_label_types = 'label_types'
gt_image_classes = 'img_cls'
gt_image_confidences = 'img_confidences'
gt_labeled_classes = 'labeled_cls'
gt_is_crowd = 'is_crowd'
gt_area = 'area'
gt_difficult = 'difficult'
gt_group_of = 'group_of'
gt_instance_masks = 'instance_masks'
gt_instance_boundaries = 'instance_boundaries'
gt_instance_classes = 'instance_classes'
image_height = 'img_height'
image_width = 'img_width'
image_size = 'img_size'
class DetectionResultFields(object):
"""Naming conventions for storing the output of the detector.
Attributes:
source_id: source of the original image.
key: unique key corresponding to image.
detection_boxes: coordinates of the detection boxes in the image.
detection_scores: detection scores for the detection boxes in the image.
detection_multiclass_scores: class score distribution (including background)
for detection boxes in the image including background class.
detection_classes: detection-level class labels.
detection_masks: contains a segmentation mask for each detection box.
"""
key = 'image_id'
detection_boxes = 'bbox'
detection_scores = 'score'
detection_classes = 'cls'
detection_masks = 'masks'
class BoxListFields(object):
"""Naming conventions for BoxLists.
Attributes:
boxes: bounding box coordinates.
classes: classes per bounding box.
scores: scores per bounding box.
weights: sample weights per bounding box.
objectness: objectness score per bounding box.
masks: masks per bounding box.
boundaries: boundaries per bounding box.
keypoints: keypoints per bounding box.
keypoint_heatmaps: keypoint heatmaps per bounding box.
is_crowd: is_crowd annotation per bounding box.
"""
boxes = 'boxes'
classes = 'classes'
scores = 'scores'
weights = 'weights'
confidences = 'confidences'
objectness = 'objectness'
masks = 'masks'
boundaries = 'boundaries'
keypoints = 'keypoints'
keypoint_visibilities = 'keypoint_visibilities'
keypoint_heatmaps = 'keypoint_heatmaps'
is_crowd = 'is_crowd'
group_of = 'group_of'
|