MCPcopy Create free account

hub / github.com/FangShancheng/ABINet-PP / functions

Functions563 in github.com/FangShancheng/ABINet-PP

↓ 73 callersMethodto
(self, device: str)
adet/structures/beziers.py:32
↓ 41 callersMethodget
(self)
demo/predictor.py:219
↓ 15 callersMethodload
(self, source, device=None, strict=True)
adet/modeling/abinet/model_vision.py:76
↓ 12 callersMethoddevice
(self)
adet/modeling/abinet/model_language.py:53
↓ 11 callersFunctionget_layer_param
(model)
adet/utils/measures.py:35
↓ 8 callersFunctionbezier_curve
adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp:22
↓ 8 callersFunctiondecoder_layer
(in_c, out_c, k=3, s=1, p=1)
adet/modeling/abinet/attention.py:13
↓ 8 callersFunctionencoder_layer
(in_c, out_c, k=3, s=2, p=1)
adet/modeling/abinet/attention.py:8
↓ 8 callersMethodoutput_shape
(self)
adet/modeling/backbone/vovnet.py:304
↓ 7 callersFunctionsplit_name
(name)
adet/modeling/backbone/bifpn.py:18
↓ 6 callersMethod__init__
(self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=2048
adet/modeling/abinet/transformer.py:479
↓ 6 callersMethoddecode
(self, rec)
adet/evaluation/text_evaluation.py:309
↓ 6 callersFunctionget_cfg
Get a copy of the default config. Returns: a detectron2 CfgNode instance.
adet/config/config.py:4
↓ 5 callersMethod__init__
(self, cfg, levels, channels, block=BasicBlock, residual_root=False)
adet/modeling/backbone/dla.py:234
↓ 5 callersFunctionconv_with_kaiming_uniform
( norm=None, activation=None, use_deformable=False, use_sep=False)
adet/layers/conv_with_kaiming_uniform.py:8
↓ 5 callersMethodpreprocess_image
Normalize, pad and batch the input images.
adet/modeling/solov2/solov2.py:148
↓ 5 callersFunctionreduce_sum
(tensor)
adet/utils/comm.py:23
↓ 4 callersMethod__init__
Args: input_ch(int) : the number of input channel out_features (list[str]): name of the layers whose outputs should
adet/modeling/backbone/vovnet.py:213
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, groups=1, norm_layer=None, filter_size=1)
adet/modeling/backbone/resnet_lpf.py:229
↓ 4 callersFunctionadd
adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp:324
↓ 4 callersFunctionbuild_optimizer
(cfg, optim, base_lr, model)
adet/utils/comm.py:143
↓ 4 callersFunctionconv1x1
1x1 convolution
adet/modeling/backbone/resnet_lpf.py:58
↓ 4 callersFunctionconv3x3
3x3 convolution with padding
adet/modeling/backbone/vovnet.py:67
↓ 4 callersFunctionconv3x3
3x3 convolution with padding
adet/modeling/backbone/resnet_lpf.py:52
↓ 4 callersFunctionget_intersection
(pD,pG)
adet/evaluation/text_eval_script.py:117
↓ 4 callersMethodput
(self, image)
demo/predictor.py:215
↓ 4 callersFunctionunfold_wo_center
(x, kernel_size, dilation)
adet/modeling/condinst/condinst.py:27
↓ 3 callersMethod__init__
Args: bottom_up (Backbone): module representing the bottom up subnetwork. Must be a subclass of :class:`Backbone`
adet/modeling/backbone/bifpn.py:286
↓ 3 callersMethod__init__
(self, cfg, in_channels)
adet/modeling/roi_heads/attn_predictor.py:30
↓ 3 callersMethod__init__
(self, cfg, input_shape: Dict[str, ShapeSpec])
adet/modeling/fcos/fcos.py:47
↓ 3 callersMethod_get_length
Greed decoder to obtain length from logit
adet/modeling/abinet/model.py:12
↓ 3 callersFunctionaligned_bilinear
(tensor, factor)
adet/utils/comm.py:38
↓ 3 callersMethoddecoder
Transform data back to its original space. In other words, return an input X_original whose transform would be X. Parameters
adet/modeling/MEInst/MaskEncoding.py:91
↓ 3 callersFunctiondice_coefficient
(x, target)
adet/modeling/condinst/dynamic_mask_head.py:51
↓ 3 callersFunctionml_nms
Performs non-maximum suppression on a boxlist, with scores specified in a boxlist field via score_field. Args: boxlist (dete
adet/layers/ml_nms.py:4
↓ 3 callersFunctionreduce_mean
(tensor)
adet/utils/comm.py:32
↓ 3 callersMethodrun
(self)
demo/predictor.py:177
↓ 2 callersMethod__init__
(self, init_value=1.0)
adet/modeling/batext/batext.py:20
↓ 2 callersMethod__init__
(self, cfg, block, layers, num_classes=1000, zero_init_residual=False, groups=1, width_per_gr
adet/modeling/backbone/resnet_lpf.py:151
↓ 2 callersMethod__init__
(self, cfg)
adet/modeling/solov2/solov2.py:33
↓ 2 callersMethod__init__
(self, conv_dim, roi_size)
adet/modeling/roi_heads/text_head.py:17
↓ 2 callersMethod__init__
(self, init_value=1.0)
adet/modeling/MEInst/MEInst.py:22
↓ 2 callersMethod__init__
Args: crop_instance (bool): if False, extend cropping boxes to avoid cropping instances
adet/data/augmentation.py:94
↓ 2 callersMethod__setstate__
(self, state)
adet/modeling/abinet/transformer.py:383
↓ 2 callersMethod_compute_index
(self, total_num_instances, max_num_instances)
adet/modeling/abinet/model_vision.py:80
↓ 2 callersFunction_get_activation_fn
(activation)
adet/modeling/abinet/transformer.py:843
↓ 2 callersFunction_get_clones
(module, N)
adet/modeling/abinet/transformer.py:839
↓ 2 callersMethod_get_location_mask
(sz, device=None)
adet/modeling/abinet/model.py:38
↓ 2 callersMethod_make_conv_level
(self, inplanes, planes, convs, stride=1, dilation=1)
adet/modeling/backbone/dla.py:288
↓ 2 callersFunctionassign_boxes_to_levels_by_metric
Map each box in `box_lists` to a feature map level index and return the assignment vector. Args: box_lists (list[detectron2.stru
adet/modeling/poolers.py:31
↓ 2 callersMethodaugment
(self, text, aug_idx)
adet/data/dataset_language.py:123
↓ 2 callersFunctionbuild_mnv2_backbone
Create a ResNet instance from config. Returns: ResNet: a :class:`ResNet` instance.
adet/modeling/backbone/mobilenet.py:139
↓ 2 callersFunctionbuild_train_loader_text
(cfg, bs=None, one_hot_y=False, num_workers=None)
adet/data/dataset_language.py:37
↓ 2 callersFunctionbuild_vovnet_backbone
Create a VoVNet instance from config. Returns: VoVNet: a :class:`VoVNet` instance.
adet/modeling/backbone/vovnet.py:314
↓ 2 callersFunctioncompute_locations
(h, w, stride, device)
adet/utils/comm.py:63
↓ 2 callersFunctionconv_layer
(in_c, out_c, k=3, s=1, p=1)
adet/modeling/abinet/feat_seq_modeling.py:9
↓ 2 callersFunctiondecode_utf8
Returns a Unicode object on success, or None on failure
adet/evaluation/rrc_evaluation_funcs.py:80
↓ 2 callersFunctiondetector_postprocess
In addition to the post processing of detectron2, we add scalign for bezier control points.
adet/modeling/abinet/model_vision.py:15
↓ 2 callersMethoddraw_instance_predictions
(self, predictions)
adet/utils/visualizer.py:25
↓ 2 callersMethodencoder
Apply dimensionality reduction to X. X is projected on the first principal components previously extracted from a training set.
adet/modeling/MEInst/MaskEncoding.py:59
↓ 2 callersMethodforward
Args: batched_inputs: a list, batched outputs of :class:`DatasetMapper` . Each item in the list contains the inpu
adet/modeling/one_stage_detector.py:76
↓ 2 callersFunctiongen_coco
(phase)
datasets/gen_coco_person.py:17
↓ 2 callersFunctiongen_crop_transform_with_instance
Generate a CropTransform so that the cropping region contains the center of the given instance. Args: crop_size (tuple): h, w in
adet/data/augmentation.py:10
↓ 2 callersMethodget_labels
Returns the labels of the corresponding text.
adet/data/dataset_language.py:112
↓ 2 callersFunctionget_tl_line_values
Validate the format of the line. If the line is not valid an exception will be raised. If maxWidth and maxHeight are specified, all points mu
adet/evaluation/rrc_evaluation_funcs.py:234
↓ 2 callersFunctionget_tl_line_values_gt
Validate the format of the line. If the line is not valid an exception will be raised. If maxWidth and maxHeight are specified, all points mu
adet/evaluation/rrc_evaluation_funcs.py:148
↓ 2 callersFunctionimresize
Resize image to a given size. Args: img (ndarray): The input image. size (tuple[int]): Target size (w, h). return_scale (b
adet/modeling/solov2/utils.py:27
↓ 2 callersMethodinitHidden
(self, batch_size)
adet/modeling/roi_heads/attn_predictor.py:97
↓ 2 callersFunctioninverse_sigmoid
Apply the inverse sigmoid operation. y = -ln(1-x/x)
adet/modeling/MEInst/LME/utils.py:14
↓ 2 callersFunctionis_leaf
(model)
adet/utils/measures.py:25
↓ 2 callersMethodload_state_dict
(self, state_dict)
adet/utils/comm.py:199
↓ 2 callersMethodmask_heads_forward_with_coords
( self, mask_feats, mask_feat_stride, instances )
adet/modeling/condinst/dynamic_mask_head.py:155
↓ 2 callersMethodmerge_bases
(self, rois, coeffs, location_to_inds=None)
adet/modeling/blendmask/blender.py:92
↓ 2 callersFunctionmulti_head_attention_forward
r""" Args: query, key, value: map a query and a set of key-value pairs to an output. See "Attention Is All You Need" for more
adet/modeling/abinet/transformer.py:15
↓ 2 callersFunctiononehot
Args: label: shape (n1, n2, ..., ) depth: a scalar Returns: onehot: (n1, n2, ..., depth)
adet/data/dataset_language.py:21
↓ 2 callersFunctionoutput
(vis, fname)
tools/visualize_data.py:61
↓ 2 callersMethodoverlay_instances
(self, beziers, recs, scores, alpha=0.15)
adet/utils/visualizer.py:79
↓ 2 callersFunctionpolygon_from_points
Returns a Polygon object to use with the Polygon2 class from a list of 8 points: x1,y1,x2,y2,x3,y3,x4,y4
adet/evaluation/text_eval_script.py:74
↓ 2 callersMethodpredict_proposals
(self, top_feats)
adet/modeling/batext/batext_outputs.py:400
↓ 2 callersMethodpredict_proposals
( self, logits_pred, reg_pred, ctrness_pred, locations, image_sizes, top_feats=None
adet/modeling/fcos/fcos_outputs.py:398
↓ 2 callersFunctionrectangle_to_polygon
(rect)
adet/evaluation/text_eval_script.py:87
↓ 2 callersFunctionrename_resnet_param_names
(ckpt_state_dict)
tools/rename_blendmask.py:24
↓ 2 callersMethodresume_or_load
(self, resume=True)
tools/train_net.py:65
↓ 2 callersMethodrun_on_video
Visualizes predictions on frames of the input video. Args: video (cv2.VideoCapture): a :class:`VideoCapture` object, who
demo/predictor.py:104
↓ 2 callersMethodtest_with_TTA
(cls, cfg, model)
tools/train_net.py:165
↓ 2 callersFunctionvalidate_clockwise_points
Validates that the points that the 4 points that dlimite a polygon are in clockwise order.
adet/evaluation/rrc_evaluation_funcs.py:298
↓ 2 callersFunctionvalidate_point_inside_bounds
(x,y,imWidth,imHeight)
adet/evaluation/rrc_evaluation_funcs.py:292
↓ 1 callersFunctionBezierAlign_backward_cpu
adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp:511
↓ 1 callersFunctionBezierAlign_forward_cpu
adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp:464
↓ 1 callersMethodMEInst_losses
( self, labels, reg_targets, logits_pred, reg_pred
adet/modeling/MEInst/MEInst_outputs.py:368
↓ 1 callersMethod__init__
Args: cfg (CfgNode): num_gpus (int): if 0, will run on CPU
demo/predictor.py:188
↓ 1 callersMethod__init__
(self, in_channels, out_channels, in_features="res5")
adet/modeling/backbone/fpn.py:40
↓ 1 callersMethod__init__
(self, cfg, n_class=1000, input_size=224, width_mult=1.)
adet/modeling/backbone/mobilenet.py:74
↓ 1 callersMethod__init__
(self, pad_type='reflect', filt_size=3, stride=2, channels=None, pad_off=0)
adet/modeling/backbone/lpf.py:9
↓ 1 callersMethod__init__
(self, d_model=256, num_layers=3, coord=False)
adet/modeling/abinet/feat_seq_modeling.py:15
↓ 1 callersMethod__init__
(self, max_length, in_channels=512, num_channels=64, mode='nearest', s_start=(1, 2), **kwarg
adet/modeling/abinet/attention.py:20
↓ 1 callersMethod__init__
(self, soft_ce=False)
adet/utils/comm.py:108
↓ 1 callersMethod__init__
(self, path:str, delimiter:str='\t', max_length:int=25,
adet/data/dataset_language.py:81
↓ 1 callersMethod__init__
(self, in_channels, out_channels, k=3)
adet/layers/gcn.py:56
↓ 1 callersMethod_bezier_to_poly
(self, bezier)
adet/utils/visualizer.py:34
next →1–100 of 563, ranked by callers