MCPcopy Create free account

hub / github.com/TropComplique/FaceBoxes-tensorflow / functions

Functions87 in github.com/TropComplique/FaceBoxes-tensorflow

↓ 4 callersMethod_add_scalewise_histograms
Adds histograms of the tensor's nonzero values for each scale (feature map). Arguments: tensor: a float tensor with shape [batch_
src/detector.py:185
↓ 4 callersFunction_float_list_feature
(value)
create_tfrecords.py:111
↓ 3 callersFunctionarea
Computes area of boxes. Arguments: boxes: a float tensor with shape [N, 4]. Returns: a float tensor with shape [N] representi
src/utils/box_utils.py:56
↓ 3 callersFunctioninception_module
(x, scope)
src/network.py:71
↓ 3 callersFunctionto_center_coordinates
Convert bounding boxes of the format [ymin, xmin, ymax, xmax] to the format [cy, cx, h, w]. Arguments: boxes: a list of float tensors
src/utils/box_utils.py:87
↓ 2 callersMethod_add_scalewise_summaries
Adds histograms of the biggest 20 percent of tensor's values for each scale (feature map). Arguments: tensor: a float ten
src/detector.py:164
↓ 2 callersFunction_bytes_feature
(value)
create_tfrecords.py:107
↓ 2 callersMethod_initialize
(self)
evaluation_utils.py:90
↓ 2 callersFunctionbatch_decode
Decodes a batch of box encodings with respect to the anchors. Arguments: box_encodings: a float tensor with shape [batch_size, num_anchor
src/utils/box_utils.py:169
↓ 2 callersFunctionget_input_fn
(is_training=True)
train.py:19
↓ 2 callersFunctionintersection
Compute pairwise intersection areas between boxes. Arguments: boxes1: a float tensor with shape [N, 4]. boxes2: a float tensor wi
src/utils/box_utils.py:30
↓ 2 callersFunctionto_minmax_coordinates
Convert bounding boxes of the format [cy, cx, h, w] to the format [ymin, xmin, ymax, xmax]. Arguments: boxes: a list of float tensors
src/utils/box_utils.py:69
↓ 1 callersMethod_add_box_predictions
Adds box predictors to each feature map, reshapes, and returns concatenated results. Arguments: feature_maps: a list of float ten
src/detector.py:240
↓ 1 callersMethod_add_scalewise_matches_summaries
Adds summaries for the number of matches on each scale.
src/detector.py:200
↓ 1 callersMethod_augmentation_fn
(self, image, boxes)
src/input_pipeline/pipeline.py:137
↓ 1 callersFunction_change_coordinate_frame
Change coordinate frame of the boxes to be relative to window's frame. Arguments: boxes: a float tensor with shape [N, 4]. window
src/input_pipeline/random_image_crop.py:149
↓ 1 callersFunction_create_targets
Returns regression targets for each anchor. Arguments: anchors: a float tensor with shape [num_anchors, 4]. groundtruth_boxes: a
src/training_target_creation.py:83
↓ 1 callersMethod_create_targets
Arguments: groundtruth: a dict with the following keys 'boxes': a float tensor with shape [batch_size, N, 4].
src/detector.py:211
↓ 1 callersFunction_ioa
Computes pairwise intersection-over-area between box collections. intersection-over-area (IOA) between two boxes box1 and box2 is defined as t
src/input_pipeline/random_image_crop.py:176
↓ 1 callersFunction_match
Matching algorithm: 1) for each groundtruth box choose the anchor with largest iou, 2) remove this set of anchors from the set of all anchors,
src/training_target_creation.py:35
↓ 1 callersFunction_prune_completely_outside_window
Prunes bounding boxes that fall completely outside of the given window. This function does not clip partially overflowing boxes. Arguments:
src/input_pipeline/random_image_crop.py:93
↓ 1 callersFunction_prune_non_overlapping_boxes
Prunes the boxes in boxes1 that overlap less than thresh with boxes2. For each box in boxes1, we want its IOA to be more than min_overlap with
src/input_pipeline/random_image_crop.py:125
↓ 1 callersFunction_random_crop_image
Performs random crop. Given the input image and its bounding boxes, this op randomly crops a subimage. Given a user-provided set of input constra
src/input_pipeline/random_image_crop.py:29
↓ 1 callersFunction_subsample_selection_to_desired_neg_pos_ratio
Subsample a collection of selected indices to a desired neg:pos ratio. Arguments: indices: an int or long tensor with shape [M],
src/losses_and_ohem.py:126
↓ 1 callersMethodadd_detections
Arguments: images: a numpy string array with shape []. boxes: a numpy float array with shape [N, 4]. scor
evaluation_utils.py:94
↓ 1 callersMethodadd_groundtruth
(self, image_name, boxes, num_boxes)
evaluation_utils.py:106
↓ 1 callersFunctionadd_weight_decay
Add L2 regularization to all (or some) trainable kernel weights.
model.py:91
↓ 1 callersFunctionapply_hard_mining
Applies hard mining to anchorwise losses. Arguments: location_losses: a float tensor with shape [batch_size, num_anchors]. cls_lo
src/losses_and_ohem.py:45
↓ 1 callersFunctionbatch_non_max_suppression
Arguments: boxes: a float tensor with shape [batch_size, N, 4]. scores: a float tensor with shape [batch_size, N]. score_
src/utils/nms.py:5
↓ 1 callersFunctionclassification_loss
Arguments: predictions: a float tensor with shape [batch_size, num_anchors, num_classes + 1], representing the predicted logi
src/losses_and_ohem.py:30
↓ 1 callersMethodclear
(self)
evaluation_utils.py:43
↓ 1 callersFunctioncompute_ap
(precision, recall)
evaluation_utils.py:236
↓ 1 callersFunctioncompute_best_threshold
Arguments: precision, recall, confidences: lists of floats of the same length. Returns: 1. a float number, best confidence t
evaluation_utils.py:177
↓ 1 callersFunctioncompute_iou
(box1, box2)
evaluation_utils.py:202
↓ 1 callersFunctioncrop
(image, boxes)
src/input_pipeline/random_image_crop.py:12
↓ 1 callersFunctiondecode
Decode relative codes to boxes. Arguments: codes: a float tensor with shape [N, 4], anchor-encoded boxes of the format [ty, t
src/utils/box_utils.py:141
↓ 1 callersFunctiondict_to_tf_example
Convert dict to tf.Example proto. Notice that this function normalizes the bounding box coordinates provided by the raw data. Arguments:
create_tfrecords.py:48
↓ 1 callersFunctionencode
Encode boxes with respect to anchors. Arguments: boxes: a float tensor with shape [N, 4]. anchors: a float tensor with shape [N,
src/utils/box_utils.py:107
↓ 1 callersMethodevaluate
(self, iou_threshold=0.5)
evaluation_utils.py:37
↓ 1 callersFunctionevaluate_detector
Arguments: groundtruth_by_img: a dict of lists with boxes, image -> list of groundtruth boxes on the image. all_detec
evaluation_utils.py:114
↓ 1 callersFunctionflip
(image, boxes)
src/input_pipeline/other_augmentations.py:41
↓ 1 callersFunctiongenerate_anchors_at_upper_left_corner
Generate densified anchor boxes at (0, 0) grid position.
src/anchor_generator.py:149
↓ 1 callersMethodget_batch
Returns: features: a dict with the following keys 'images': a float tensor with shape [batch_size, image_height,
src/input_pipeline/pipeline.py:73
↓ 1 callersMethodget_metric_ops
Arguments: image_name: a string tensor with shape [1]. groundtruth: a dict with the following keys 'b
evaluation_utils.py:46
↓ 1 callersMethodget_predictions
Postprocess outputs of the network. Returns: boxes: a float tensor with shape [batch_size, N, 4]. scores: a float ten
src/detector.py:59
↓ 1 callersFunctionget_training_targets
Arguments: anchors: a float tensor with shape [num_anchors, 4]. groundtruth_boxes: a float tensor with shape [N, 4]. thre
src/training_target_creation.py:5
↓ 1 callersFunctioniou
Computes pairwise intersection-over-union between two box collections. Arguments: boxes1: a float tensor with shape [N, 4]. boxes
src/utils/box_utils.py:13
↓ 1 callersFunctionlocalization_loss
A usual L1 smooth loss. Arguments: predictions: a float tensor with shape [batch_size, num_anchors, 4], representing the (enc
src/losses_and_ohem.py:11
↓ 1 callersMethodloss
Compute scalar loss tensors with respect to provided groundtruth. Arguments: groundtruth: a dict with the following keys
src/detector.py:86
↓ 1 callersFunctionmain
()
create_pb.py:18
↓ 1 callersFunctionmain
()
create_tfrecords.py:115
↓ 1 callersFunctionmake_args
()
create_pb.py:7
↓ 1 callersFunctionmake_args
()
create_tfrecords.py:39
↓ 1 callersFunctionmanipulate
(image)
src/input_pipeline/other_augmentations.py:13
↓ 1 callersFunctionmatch
Arguments: detection: a box. groundtruth_boxes: a list of boxes. Returns: best_i: an integer, index of the best groun
evaluation_utils.py:217
↓ 1 callersFunctionpreprocess
Transform images before feeding them to the network.
src/network.py:66
↓ 1 callersFunctionrandom_color_manipulations
(image, probability=0.5, grayscale_probability=0.1)
src/input_pipeline/other_augmentations.py:11
↓ 1 callersFunctionrandom_flip_left_right
(image, boxes)
src/input_pipeline/other_augmentations.py:39
↓ 1 callersFunctionrandom_image_crop
( image, boxes, probability=0.5, min_object_covered=0.9, aspect_ratio_range=(0.75, 1.3
src/input_pipeline/random_image_crop.py:5
↓ 1 callersFunctionrandom_jitter_box
Randomly jitter a box. Arguments: box: a float tensor with shape [4]. ratio: a float number. Returns:
src/input_pipeline/other_augmentations.py:94
↓ 1 callersFunctionrandom_jitter_boxes
Randomly jitter bounding boxes. Arguments: boxes: a float tensor with shape [N, 4]. ratio: a float number. The ratio
src/input_pipeline/other_augmentations.py:82
↓ 1 callersFunctionrandom_pixel_value_scale
This function scales each pixel independently of the other ones. Arguments: image: a float tensor with shape [height, width, 3],
src/input_pipeline/other_augmentations.py:55
↓ 1 callersFunctionrandom_value_scale
(image)
src/input_pipeline/other_augmentations.py:67
↓ 1 callersFunctiontile_anchors
Arguments: image_size: a tuple of integers (width, height). grid_height: an integer, size of the grid in the y direction.
src/anchor_generator.py:102
↓ 1 callersFunctionto_grayscale
(image)
src/input_pipeline/other_augmentations.py:23
Method__call__
Detect faces. Arguments: image: a numpy uint8 array with shape [height, width, 3], that represents a RGB image.
face_detector.py:35
Method__call__
Arguments: images: a float tensor with shape [batch_size, height, width, 3], a batch of RGB images with pixel val
src/network.py:10
Method__call__
Arguments: image_features: a list of float tensors where the ith tensor has shape [batch, height_i, width_i, chan
src/anchor_generator.py:40
Method__init__
Arguments: image: a string, identifier of a image. box: a numpy float array with shape [4]. score: a floa
evaluation_utils.py:13
Method__init__
(self)
evaluation_utils.py:34
Method__init__
Arguments: model_path: a string, path to a pb file. gpu_memory_fraction: a float number. visible_device_l
face_detector.py:6
Method__init__
(self, is_training)
src/network.py:7
Method__init__
Arguments: images: a float tensor with shape [batch_size, height, width, 3], a batch of RGB images with pixel val
src/detector.py:12
Method__init__
(self)
src/anchor_generator.py:37
Method__init__
Note: when evaluating set batch_size to 1. Arguments: filenames: a list of strings, paths to tfrecords files.
src/input_pipeline/pipeline.py:12
Method_parse_and_preprocess
What this function does: 1. Parses one record from a tfrecords file and decodes it. 2. (optionally) Augments it. Returns:
src/input_pipeline/pipeline.py:89
Methodbatch_norm
(x)
src/network.py:20
Methodevaluate_func
()
evaluation_utils.py:69
Functionfn
(x)
src/utils/nms.py:21
Methodfn
(x)
src/detector.py:221
Methodget_num_samples
(filename)
src/input_pipeline/pipeline.py:39
Methodget_value_func
(measure)
evaluation_utils.py:74
Functioninput_fn
()
train.py:31
Functionmodel_fn
This is a function for creating a computational tensorflow graph. The function is in format required by tf.estimator.
model.py:9
Functionserving_input_receiver_fn
()
save.py:33
Methodupdate_op_func
(image_name, gt_boxes, gt_num_boxes, boxes, scores, num_boxes)
evaluation_utils.py:59
Methodvalue_func
()
evaluation_utils.py:75