MCPcopy Create free account

hub / github.com/FangShancheng/ABINet-PP / types & classes

Types & classes108 in github.com/FangShancheng/ABINet-PP

↓ 10 callersClassDLA
adet/modeling/backbone/dla.py:233
↓ 6 callersClassDownsample
adet/modeling/backbone/lpf.py:8
↓ 6 callersClassTree
adet/modeling/backbone/dla.py:175
↓ 5 callersClassPositionalEncoding
r"""Inject some information about the relative or absolute position of the tokens in the sequence. The positional encodings have the same dime
adet/modeling/abinet/transformer.py:852
↓ 4 callersClassBeziers
This structure stores a list of bezier curves as a Nx16 torch.Tensor. It will support some common methods about bezier shapes (`area`, `c
adet/structures/beziers.py:5
↓ 4 callersClassConv2D
adet/layers/gcn.py:6
↓ 4 callersClassMultiheadAttention
r"""Allows the model to jointly attend to information from different representation subspaces. See reference: Attention Is All You Need ..
adet/modeling/abinet/transformer.py:299
↓ 3 callersClassCrossEntropyLoss
adet/utils/comm.py:107
↓ 3 callersClassIOULoss
Intersetion Over Union (IoU) loss which supports three different IoU computations: * IoU * Linear IoU * gIoU
adet/layers/iou_loss.py:5
↓ 3 callersClassLastLevelP6
This module is used in FCOS to generate extra layers
adet/modeling/backbone/fpn.py:35
↓ 3 callersClassLastLevelP6P7
This module is used in RetinaNet and FCOS to generate extra layers, P6 and P7 from C5 or P5 feature.
adet/modeling/backbone/fpn.py:14
↓ 3 callersClassTransformer
r"""A transformer model. User is able to modify the attributes as needed. The architecture is based on the paper "Attention Is All You Need". Ashi
adet/modeling/abinet/transformer.py:450
↓ 2 callersClassABIAlignment
adet/modeling/abinet/model_alignment.py:8
↓ 2 callersClassABILanguage
adet/modeling/abinet/model_language.py:16
↓ 2 callersClassABIVision
Same as :class:`detectron2.modeling.ProposalNetwork`. Use one stage detector and a second stage for instance-wise prediction.
adet/modeling/abinet/model_vision.py:56
↓ 2 callersClassAdetCheckpointer
Same as :class:`DetectronCheckpointer`, but is able to convert models in AdelaiDet, such as LPF backbone.
adet/checkpoint/adet_checkpoint.py:6
↓ 2 callersClassDatasetMapperWithBasis
This caller enables the default Detectron2 mapper to read an additional basis semantic label
adet/data/dataset_mapper.py:55
↓ 2 callersClassMaskLoader
Dataloader for Local Mask. Arguments: root (string): filepath to dataset folder. dataset (string): mask to use (eg. 'train',
adet/modeling/MEInst/LME/MaskLoader.py:28
↓ 2 callersClassModuleListDial
adet/modeling/fcos/fcos.py:29
↓ 2 callersClassNaiveGroupNorm
r"""NaiveGroupNorm implements Group Normalization with the high-level matrix operations in PyTorch. It is a temporary solution to export GN by ONN
adet/layers/naive_group_norm.py:6
↓ 2 callersClassResNet
adet/modeling/abinet/feat_seq_modeling.py:14
↓ 2 callersClassTextVisualizer
adet/utils/visualizer.py:13
↓ 2 callersClassTopPooler
ROIPooler with option to assign level by max length. Used by top modules.
adet/modeling/poolers.py:83
↓ 2 callersClassTransformerDecoder
r"""TransformerDecoder is a stack of N decoder layers Args: decoder_layer: an instance of the TransformerDecoderLayer() class (required).
adet/modeling/abinet/transformer.py:625
↓ 2 callersClassTransformerDecoderLayer
r"""TransformerDecoderLayer is made up of self-attn, multi-head-attn and feedforward network. This standard decoder layer is based on the paper "A
adet/modeling/abinet/transformer.py:746
↓ 2 callersClassTransformerEncoder
r"""TransformerEncoder is a stack of N encoder layers Args: encoder_layer: an instance of the TransformerEncoderLayer() class (required).
adet/modeling/abinet/transformer.py:580
↓ 2 callersClassTransformerEncoderLayer
r"""TransformerEncoderLayer is made up of self-attn and feedforward network. This standard encoder layer is based on the paper "Attention Is All Y
adet/modeling/abinet/transformer.py:679
↓ 2 callersClass_OSA_module
adet/modeling/backbone/vovnet.py:125
↓ 1 callersClassATTPredictor
adet/modeling/roi_heads/attn_predictor.py:107
↓ 1 callersClassAsyncPredictor
A predictor that runs the model asynchronously, possibly on >1 GPUs. Because rendering the visualization takes considerably amount of time,
demo/predictor.py:160
↓ 1 callersClassAttention
adet/modeling/roi_heads/attn_predictor.py:49
↓ 1 callersClassAutonomousOptimizer
adet/utils/comm.py:178
↓ 1 callersClassBATextOutputs
adet/modeling/batext/batext_outputs.py:125
↓ 1 callersClassBackboneWithTopLevels
adet/modeling/backbone/bifpn.py:57
↓ 1 callersClassBezierAlign
adet/layers/bezier_align.py:51
↓ 1 callersClassBiFPN
This module implements Feature Pyramid Network. It creates pyramid features built on top of some input feature maps.
adet/modeling/backbone/bifpn.py:280
↓ 1 callersClassBidirectionalLSTM
adet/modeling/roi_heads/attn_predictor.py:10
↓ 1 callersClassBlender
adet/modeling/blendmask/blender.py:12
↓ 1 callersClassCRNN
adet/modeling/roi_heads/attn_predictor.py:29
↓ 1 callersClassDynamicMaskHead
adet/modeling/condinst/dynamic_mask_head.py:94
↓ 1 callersClassFCOSHead
adet/modeling/batext/batext.py:170
↓ 1 callersClassFCOSHead
adet/modeling/fcos/fcos.py:122
↓ 1 callersClassFCOSOutputs
adet/modeling/fcos/fcos_outputs.py:52
↓ 1 callersClassFeatureMapResampler
adet/modeling/backbone/bifpn.py:25
↓ 1 callersClassGCN
Large Kernel Matters -- https://arxiv.org/abs/1703.02719
adet/layers/gcn.py:52
↓ 1 callersClassHsigmoid
adet/modeling/backbone/vovnet.py:100
↓ 1 callersClassIOUMetric
Class to calculate mean-iou using fast_hist method
adet/modeling/MEInst/LME/utils.py:92
↓ 1 callersClassMEInstHead
adet/modeling/MEInst/MEInst.py:179
↓ 1 callersClassMEInstOutputs
adet/modeling/MEInst/MEInst_outputs.py:57
↓ 1 callersClassMaskBranch
adet/modeling/condinst/mask_branch.py:21
↓ 1 callersClassMobileNetV2
Should freeze bn
adet/modeling/backbone/mobilenet.py:70
↓ 1 callersClassPCAMaskEncoding
To do the mask encoding of PCA. components_: (tensor), shape (n_components, n_features) if agnostic=True
adet/modeling/MEInst/MaskEncoding.py:10
↓ 1 callersClassPositionAttention
adet/modeling/abinet/attention.py:19
↓ 1 callersClassPositionContentAttention
adet/modeling/abinet/attention.py:81
↓ 1 callersClassRNNPredictor
adet/modeling/roi_heads/text_head.py:38
↓ 1 callersClassRandomCropWithInstanceAR
Instance-aware cropping.
adet/data/augmentation.py:110
↓ 1 callersClassRandomRotationWithProb
adet/data/augmentation.py:140
↓ 1 callersClassResNetLPF
adet/modeling/backbone/resnet_lpf.py:149
↓ 1 callersClassRoot
adet/modeling/backbone/dla.py:154
↓ 1 callersClassSOLOv2InsHead
adet/modeling/solov2/solov2.py:507
↓ 1 callersClassSOLOv2MaskHead
adet/modeling/solov2/solov2.py:631
↓ 1 callersClassScale
adet/modeling/batext/batext.py:19
↓ 1 callersClassScale
adet/modeling/fcos/fcos.py:20
↓ 1 callersClassScale
adet/modeling/MEInst/MEInst.py:21
↓ 1 callersClassSeqConvs
adet/modeling/roi_heads/text_head.py:16
↓ 1 callersClassSingleBiFPN
This module implements Feature Pyramid Network. It creates pyramid features built on top of some input feature maps.
adet/modeling/backbone/bifpn.py:101
↓ 1 callersClassSoftCrossEntropyLoss
adet/utils/comm.py:94
↓ 1 callersClassSpellingMutation
adet/data/dataset_language.py:174
↓ 1 callersClassTextDataset
adet/data/dataset_language.py:80
↓ 1 callersClassTextEvaluator
Evaluate text proposals and recognition.
adet/evaluation/text_evaluation.py:27
↓ 1 callersClassTextIterator
adet/data/dataset_language.py:64
↓ 1 callersClassTrainer
This is the same Trainer except that we rewrite the `build_train_loader`/`resume_or_load` method.
tools/train_net.py:48
↓ 1 callersClassVisualizationDemo
demo/predictor.py:19
↓ 1 callersClassVoVNet
adet/modeling/backbone/vovnet.py:211
↓ 1 callersClass_OSA_stage
adet/modeling/backbone/vovnet.py:172
↓ 1 callersClasseSEModule
adet/modeling/backbone/vovnet.py:109
ClassABINetIterModel
adet/modeling/abinet/model_abinet_iter.py:11
ClassABINetModel
adet/modeling/abinet/model_abinet.py:11
ClassBAText
A modified version of FCOS with Bezier regression
adet/modeling/batext/batext.py:29
ClassBasicBlock
adet/modeling/backbone/dla.py:38
ClassBasicBlock
adet/modeling/backbone/resnet_lpf.py:63
ClassBlendMask
Main class for BlendMask architectures (see https://arxiv.org/abd/1901.02446).
adet/modeling/blendmask/blendmask.py:22
ClassBottleneck
adet/modeling/backbone/dla.py:69
ClassBottleneck
adet/modeling/backbone/resnet_lpf.py:104
ClassBottleneckX
adet/modeling/backbone/dla.py:110
ClassCondInst
Main class for CondInst architectures (see https://arxiv.org/abs/2003.05664).
adet/modeling/condinst/condinst.py:74
ClassDFConv2d
Deformable convolutional layer with configurable deformable groups, dilations and groups. Code is from: https://github.com/facebookr
adet/layers/deform_conv.py:19
ClassDefROIAlign
adet/layers/def_roi_align.py:60
ClassDownsample1D
adet/modeling/backbone/lpf.py:62
ClassFCOS
Implement FCOS (https://arxiv.org/abs/1904.01355).
adet/modeling/fcos/fcos.py:43
ClassFocalLoss
adet/modeling/solov2/loss.py:91
ClassInvertedResidual
adet/modeling/backbone/mobilenet.py:29
ClassMEInst
Implement MEInst (https://arxiv.org/abs/2003.11712).
adet/modeling/MEInst/MEInst.py:31
ClassModel
adet/modeling/abinet/model.py:5
ClassOneStageDetector
Same as :class:`detectron2.modeling.ProposalNetwork`. Uses "instances" as the return key instead of using "proposal".
adet/modeling/one_stage_detector.py:39
ClassOneStageRCNN
Same as :class:`detectron2.modeling.ProposalNetwork`. Use one stage detector and a second stage for instance-wise prediction.
adet/modeling/one_stage_detector.py:66
ClassPCATextHead
TextHead performs text region alignment and recognition. It is a simplified ROIHeads, only ground truth RoIs are used during trainin
adet/modeling/abinet/attn_text_head.py:17
ClassPreCalc
adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp:9
ClassProtoNet
adet/modeling/blendmask/basis_module.py:26
ClassRandomCropWithInstance
Instance-aware cropping.
adet/data/augmentation.py:90
next →1–100 of 108, ranked by callers