MCPcopy Create free account

hub / github.com/610265158/DSFD-tensorflow / functions

Functions181 in github.com/610265158/DSFD-tensorflow

↓ 20 callersFunctionresnet_v1_block
Helper function for creating a resnet_v1 bottleneck block. Args: scope: The scope of the block. base_depth: The depth of the bottleneck laye
lib/core/model/net/resnet/resnet_v1.py:238
↓ 18 callersFunctionop
(opfunc, multiplier_func=depth_multiplier, **params)
lib/core/model/net/mobilenet/mobilenet.py:107
↓ 16 callersFunctionresnet_v2_block
Helper function for creating a resnet bottleneck block. Args: scope: The scope of the block. base_depth: The depth of the bottleneck layer f
lib/core/model/net/resnet/resnet_v2.py:211
↓ 8 callersMethodadd_summary
(self,event)
lib/core/base_trainer/net_work.py:115
↓ 6 callersFunctiondetect_face
(img, shrink)
model_eval/wider.py:80
↓ 6 callersFunctionl2_normalization
(x, scale, name)
lib/core/model/net/vgg/backbone.py:10
↓ 6 callersFunctionresnet_arg_scope
(bn_is_training, bn_trainable=True, trainable=True,
lib/core/model/net/arg_scope/resnet_args_cope.py:14
↓ 5 callersFunctionresnet_v1
Generator for v1 ResNet models. This function generates a family of ResNet v1 models. See the resnet_v1_*() methods for specific model instantiati
lib/core/model/net/resnet/resnet_v1.py:128
↓ 4 callersFunctionFill_img
(img_raw,target_height,target_width,label=None)
lib/dataset/augmentor/augmentation.py:187
↓ 4 callersFunction_fixed_padding
Pads the input along the spatial dimensions independently of input size. Pads the input such that if it was used in a convolution with 'VALID' paddi
lib/core/model/net/mobilenet/conv_blocks.py:24
↓ 4 callersFunction_uniform
Uniformly sample from the given range. Args val_range: A pair of lower and upper bound.
lib/dataset/augmentor/visual_augmentation.py:126
↓ 4 callersFunctionresnet_v2
Generator for v2 (preactivation) ResNet models. This function generates a family of ResNet v2 models. See the resnet_v2_*() methods for specific m
lib/core/model/net/resnet/resnet_v2.py:101
↓ 4 callersFunctionssd_loss
(reg_predict,cla_predict,reg_label,cla_label,which_loss='focal_loss')
lib/core/model/net/ssd_loss.py:10
↓ 3 callersFunctionRandom_scale_withbbox
(image,bboxes,target_shape,jitter=0.5)
lib/dataset/augmentor/augmentation.py:256
↓ 3 callersFunctioncreate_fem_net
(blocks, L2_reg,is_training, trainable=True,data_format='NHWC')
lib/core/model/net/FEM.py:31
↓ 3 callersMethodgenerate_cell_anchor
(self,base_size=16,ratios=[0.5,1.,2.],scales=2**np.arange(3,6),rect=cfg.ANCHOR.rect)
lib/core/anchor/anchor.py:21
↓ 3 callersFunctionsafe_arg_scope
Returns `slim.arg_scope` with all None arguments removed. Arguments: funcs: Functions to pass to `arg_scope`. **kwargs: Arguments to pass to
lib/core/model/net/mobilenet/mobilenet.py:122
↓ 3 callersFunctionwrapped_partial
(func, *args, **kwargs)
lib/core/model/net/mobilenet/mobilenet_v2.py:162
↓ 2 callersFunctionRandom_flip
(im, boxes)
lib/dataset/augmentor/augmentation.py:307
↓ 2 callersFunction_clip
Clip and convert an image to np.uint8. Args image: Image to clip.
lib/dataset/augmentor/visual_augmentation.py:118
↓ 2 callersFunction_set_arg_scope_defaults
Sets arg scope defaults for all items present in defaults. Args: defaults: dictionary/list of pairs, containing a mapping from function to a
lib/core/model/net/mobilenet/mobilenet.py:70
↓ 2 callersFunction_split_divisible
Evenly splits num, num_ways so each piece is a multiple of divisible_by.
lib/core/model/net/mobilenet/conv_blocks.py:57
↓ 2 callersMethod_to_whxy
(self,anchors)
lib/core/anchor/anchor.py:28
↓ 2 callersMethod_to_xyxy
(self,w,h,x,y)
lib/core/anchor/anchor.py:37
↓ 2 callersFunctionarea
Computes area of boxes. Arguments: boxes: a float tensor with shape [N, 4]. Returns: a float tensor with shape [N] representi
lib/core/anchor/box_utils.py:16
↓ 2 callersFunctionget_all_anchors_fpn
Returns: [anchors]: each anchors is a SxSx NUM_ANCHOR_RATIOS x4 array.
lib/core/anchor/tf_anchors.py:77
↓ 2 callersFunctionparse_wider_file
(root, file)
prepare_wider_data.py:22
↓ 2 callersMethodproduce_anchors
(self)
lib/core/anchor/anchor.py:96
↓ 2 callersFunctionsplit_conv
Creates a split convolution. Split convolution splits the input and output into 'num_blocks' blocks of approximately the same size each, and onl
lib/core/model/net/mobilenet/conv_blocks.py:311
↓ 2 callersFunctiontraining_scope
Defines Mobilenet training scope. Usage: with tf.contrib.slim.arg_scope(mobilenet.training_scope()): logits, endpoints = mobilenet_v2.mo
lib/core/model/net/mobilenet/mobilenet.py:401
↓ 1 callersMethodFill_img
(self,img_raw,target_height,target_width,label=None)
lib/core/api/face_detector.py:78
↓ 1 callersFunctionGetFileList
(dir, fileList)
vis.py:14
↓ 1 callersFunctionRotate_coordinate
(label,rt_matrix)
lib/dataset/augmentor/augmentation.py:41
↓ 1 callersMethod__init__
(self, img_root_path='', ann_file=None, training_flag=True)
lib/dataset/dataietr.py:304
↓ 1 callersFunction_fixed_padding
Pads the input along the spatial dimensions independently of input size. Pads the input such that if it was used in a convolution with 'VALID' paddi
lib/core/model/net/mobilenet/mobilenet.py:36
↓ 1 callersFunction_has_no_variables
Determines if the graph has any variables. Args: sess: TensorFlow Session. Returns: Bool.
tools/freeze.py:57
↓ 1 callersFunction_make_divisible
(v, divisor, min_value=None)
lib/core/model/net/mobilenet/mobilenet.py:59
↓ 1 callersFunction_make_divisible
(v, divisor, min_value=None)
lib/core/model/net/mobilenet/conv_blocks.py:47
↓ 1 callersMethod_map_func
Data augmentation function.
lib/dataset/dataietr.py:213
↓ 1 callersFunction_parse_input_graph_proto
Parses input tensorflow graph into GraphDef proto.
tools/freeze.py:234
↓ 1 callersFunction_parse_input_meta_graph_proto
Parses input tensorflow graph into MetaGraphDef proto.
tools/freeze.py:248
↓ 1 callersFunction_parse_input_saver_proto
Parses input tensorflow Saver into SaverDef proto.
tools/freeze.py:263
↓ 1 callersFunction_scope_all
(scope, default_scope=None)
lib/core/model/net/mobilenet/mobilenet.py:291
↓ 1 callersMethod_train
(self,_epoch)
lib/core/base_trainer/net_work.py:364
↓ 1 callersFunction_v1_compatible_scope_naming
(scope)
lib/core/model/net/mobilenet/conv_blocks.py:76
↓ 1 callersMethod_val
(self,_epoch)
lib/core/base_trainer/net_work.py:447
↓ 1 callersFunctionadjust_brightness
Adjust brightness of an image Args image: Image to adjust. delta: Brightness offset between -1 and 1 added to the pixel values.
lib/dataset/augmentor/visual_augmentation.py:85
↓ 1 callersFunctionadjust_contrast
Adjust contrast of an image. Args image: Image to adjust. factor: A factor for adjusting contrast.
lib/dataset/augmentor/visual_augmentation.py:74
↓ 1 callersFunctionadjust_hue
Adjust hue of an image. Args image: Image to adjust. delta: An interval between -1 and 1 for the amount added to the hue channel
lib/dataset/augmentor/visual_augmentation.py:95
↓ 1 callersFunctionadjust_saturation
Adjust saturation of an image. Args image: Image to adjust. factor: An interval for the factor multiplying the saturation values
lib/dataset/augmentor/visual_augmentation.py:107
↓ 1 callersMethodaverage_gradients
Calculate the average gradient for each shared variable across all towers. Note that this function provides a synchronization point across al
lib/core/base_trainer/net_work.py:139
↓ 1 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
lib/core/anchor/box_utils.py:130
↓ 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_
lib/core/anchor/nms.py:4
↓ 1 callersFunctionbbox_vote
(det)
model_eval/wider.py:45
↓ 1 callersFunctionbbox_vote
(det)
model_eval/fddb.py:63
↓ 1 callersFunctionbox_in_img
(img,boxes,min_overlap=0.5)
lib/dataset/augmentor/augmentation.py:228
↓ 1 callersFunctionbox_to_point
:param boxes: [n,x,y,x,y] :return: [4n,x,y]
lib/dataset/augmentor/augmentation.py:50
↓ 1 callersMethodbuild
(self)
lib/core/base_trainer/net_work.py:181
↓ 1 callersMethodbuild_iter
(self,)
lib/dataset/dataietr.py:324
↓ 1 callersFunctioncpm
(product,scope)
lib/core/model/net/FEM.py:8
↓ 1 callersFunctiondecode
Decode relative codes to boxes. Arguments: codes: a float tensor with shape [N, 4], anchor-encoded boxes of the format [tx1, t
lib/core/anchor/box_utils.py:100
↓ 1 callersFunctionencode
Encode boxes with respect to anchors. Arguments: boxes: a float tensor with shape [N, 4]. anchors: a float tensor with shape [N, 4
lib/core/anchor/box_utils.py:71
↓ 1 callersFunctionexpand_input_by_factor
(n, divisible_by=8)
lib/core/model/net/mobilenet/conv_blocks.py:152
↓ 1 callersFunctionextra_feature
(x)
lib/core/model/net/vgg/backbone.py:15
↓ 1 callersFunctionfacedetect
()
vis.py:27
↓ 1 callersFunctionflip_test
(image, shrink)
model_eval/wider.py:131
↓ 1 callersFunctionfocal_loss
r"""Compute focal loss for predictions. Multi-labels Focal loss formula: FL = -alpha * (z-p)^gamma * log(p) -(1-alpha) * p^gamma *
lib/core/model/net/ssd_loss.py:88
↓ 1 callersMethodforward
(self,inputs,boxes,labels,l2_regulation,training_flag,with_loss=True)
lib/core/model/net/ssd.py:42
↓ 1 callersFunctionfreeze_graph
Converts all variables in a graph and checkpoint into constants. Args: input_graph: A `GraphDef` file to load. input_saver: A TensorFlow Sav
tools/freeze.py:277
↓ 1 callersFunctionfreeze_graph_with_def_protos
Converts all variables in a graph and checkpoint into constants. Args: input_graph_def: A `GraphDef`. input_saver_def: A `SaverDef` (optiona
tools/freeze.py:70
↓ 1 callersMethodfrozen
(self)
lib/core/base_trainer/net_work.py:86
↓ 1 callersFunctionget_all_anchors
Get all anchors in the largest possible image, shifted, floatbox Args: max_size(int) : h w stride (int): the stride of anchor
lib/core/anchor/tf_anchors.py:12
↓ 1 callersMethodget_all_anchors
Get all anchors in the largest possible image, shifted, floatbox Args: stride (int): the stride of anchors. s
lib/core/anchor/anchor.py:102
↓ 1 callersMethodget_all_anchors_fpn
Returns: [anchors]: each anchors is a SxSx NUM_ANCHOR_RATIOS x4 array.
lib/core/anchor/anchor.py:149
↓ 1 callersMethodget_all_sample
(self)
lib/dataset/dataietr.py:54
↓ 1 callersMethodget_anchor_labels
(self,anchors, gt_boxes, labels)
lib/core/anchor/anchor.py:193
↓ 1 callersFunctionget_data
()
model_eval/wider.py:27
↓ 1 callersFunctionget_logger
(LEVEL,log_file=None)
lib/helper/logger.py:11
↓ 1 callersMethodget_opt
(self)
lib/core/base_trainer/net_work.py:39
↓ 1 callersFunctionglobal_pool
Applies avg pool to produce 1x1 output. NOTE: This function is funcitonally equivalenet to reduce_mean, but it has baked in average pool which has
lib/core/model/net/mobilenet/mobilenet.py:377
↓ 1 callersMethodinit_model
(self,args)
lib/core/api/face_detector.py:104
↓ 1 callersMethodintersect
(self,box_a, box_b)
lib/dataset/augmentor/augmentation.py:491
↓ 1 callersMethodintersect
(self,box_a, box_b)
lib/dataset/augmentor/augmentation.py:623
↓ 1 callersFunctionintersection
Compute pairwise intersection areas between boxes. Arguments: boxes1: a float tensor with shape [N, 4]. boxes2: a float tensor with sha
lib/core/anchor/box_utils.py:44
↓ 1 callersMethodjaccard_numpy
Compute the jaccard overlap of two sets of boxes. The jaccard overlap is simply the intersection over union of two boxes. Args:
lib/dataset/augmentor/augmentation.py:473
↓ 1 callersMethodjaccard_numpy
Compute the jaccard overlap of two sets of boxes. The jaccard overlap is simply the intersection over union of two boxes. Args:
lib/dataset/augmentor/augmentation.py:605
↓ 1 callersMethodload_weight
(self)
lib/core/base_trainer/net_work.py:55
↓ 1 callersFunctionlocalization_loss
A usual L1 smooth loss. Arguments: predictions: a float tensor with shape [batch_size, num_anchors, 4], representing the (enc
lib/core/model/net/ssd_loss.py:69
↓ 1 callersFunctionmain
(unused_args, flags)
tools/freeze.py:354
↓ 1 callersMethodmake_anchor_in_ratios
(self,base_anchor,ratios,rect=False)
lib/core/anchor/anchor.py:47
↓ 1 callersMethodmake_anchor_in_sclaes
(self,anchors,scales)
lib/core/anchor/anchor.py:70
↓ 1 callersMethodmax_out_cla
(self,fm,ratios_per_pixel,cla_num=2,scope='nul')
lib/core/model/net/ssd_head.py:14
↓ 1 callersFunctionmobilenet
Creates mobilenet V2 network. Inference mode is created by default. To create training use training_scope below. with tf.contrib.slim.arg_scope(
lib/core/model/net/mobilenet/mobilenet_v2.py:84
↓ 1 callersFunctionmobilenet_base
Mobilenet base network. Constructs a network from inputs to the given final endpoint. By default the network is constructed in inference mode. To
lib/core/model/net/mobilenet/mobilenet.py:141
↓ 1 callersFunctionmulti_scale_test
( image, max_im_shrink)
model_eval/wider.py:98
↓ 1 callersFunctionnp_iou
(boxes1, boxes2)
lib/core/anchor/box_utils.py:15
↓ 1 callersFunctionohem_loss
(logits, targets, weights)
lib/core/model/net/ssd_loss.py:122
↓ 1 callersMethodparse_file
:return: [fname,lbel] type:list
lib/dataset/dataietr.py:202
↓ 1 callersFunctionpoint_to_box
(points)
lib/dataset/augmentor/augmentation.py:69
↓ 1 callersMethodpostprocess
Postprocess outputs of the network. Returns: boxes: a float tensor with shape [batch_size, N, 4]. scores: a float ten
lib/core/model/net/ssd.py:129
next →1–100 of 181, ranked by callers