Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/FangShancheng/ABINet-PP
/ types & classes
Types & classes
108 in github.com/FangShancheng/ABINet-PP
⨍
Functions
563
◇
Types & classes
108
↓ 10 callers
Class
DLA
adet/modeling/backbone/dla.py:233
↓ 6 callers
Class
Downsample
adet/modeling/backbone/lpf.py:8
↓ 6 callers
Class
Tree
adet/modeling/backbone/dla.py:175
↓ 5 callers
Class
PositionalEncoding
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 callers
Class
Beziers
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 callers
Class
Conv2D
adet/layers/gcn.py:6
↓ 4 callers
Class
MultiheadAttention
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 callers
Class
CrossEntropyLoss
adet/utils/comm.py:107
↓ 3 callers
Class
IOULoss
Intersetion Over Union (IoU) loss which supports three different IoU computations: * IoU * Linear IoU * gIoU
adet/layers/iou_loss.py:5
↓ 3 callers
Class
LastLevelP6
This module is used in FCOS to generate extra layers
adet/modeling/backbone/fpn.py:35
↓ 3 callers
Class
LastLevelP6P7
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 callers
Class
Transformer
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 callers
Class
ABIAlignment
adet/modeling/abinet/model_alignment.py:8
↓ 2 callers
Class
ABILanguage
adet/modeling/abinet/model_language.py:16
↓ 2 callers
Class
ABIVision
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 callers
Class
AdetCheckpointer
Same as :class:`DetectronCheckpointer`, but is able to convert models in AdelaiDet, such as LPF backbone.
adet/checkpoint/adet_checkpoint.py:6
↓ 2 callers
Class
DatasetMapperWithBasis
This caller enables the default Detectron2 mapper to read an additional basis semantic label
adet/data/dataset_mapper.py:55
↓ 2 callers
Class
MaskLoader
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 callers
Class
ModuleListDial
adet/modeling/fcos/fcos.py:29
↓ 2 callers
Class
NaiveGroupNorm
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 callers
Class
ResNet
adet/modeling/abinet/feat_seq_modeling.py:14
↓ 2 callers
Class
TextVisualizer
adet/utils/visualizer.py:13
↓ 2 callers
Class
TopPooler
ROIPooler with option to assign level by max length. Used by top modules.
adet/modeling/poolers.py:83
↓ 2 callers
Class
TransformerDecoder
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 callers
Class
TransformerDecoderLayer
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 callers
Class
TransformerEncoder
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 callers
Class
TransformerEncoderLayer
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 callers
Class
_OSA_module
adet/modeling/backbone/vovnet.py:125
↓ 1 callers
Class
ATTPredictor
adet/modeling/roi_heads/attn_predictor.py:107
↓ 1 callers
Class
AsyncPredictor
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 callers
Class
Attention
adet/modeling/roi_heads/attn_predictor.py:49
↓ 1 callers
Class
AutonomousOptimizer
adet/utils/comm.py:178
↓ 1 callers
Class
BATextOutputs
adet/modeling/batext/batext_outputs.py:125
↓ 1 callers
Class
BackboneWithTopLevels
adet/modeling/backbone/bifpn.py:57
↓ 1 callers
Class
BezierAlign
adet/layers/bezier_align.py:51
↓ 1 callers
Class
BiFPN
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 callers
Class
BidirectionalLSTM
adet/modeling/roi_heads/attn_predictor.py:10
↓ 1 callers
Class
Blender
adet/modeling/blendmask/blender.py:12
↓ 1 callers
Class
CRNN
adet/modeling/roi_heads/attn_predictor.py:29
↓ 1 callers
Class
DynamicMaskHead
adet/modeling/condinst/dynamic_mask_head.py:94
↓ 1 callers
Class
FCOSHead
adet/modeling/batext/batext.py:170
↓ 1 callers
Class
FCOSHead
adet/modeling/fcos/fcos.py:122
↓ 1 callers
Class
FCOSOutputs
adet/modeling/fcos/fcos_outputs.py:52
↓ 1 callers
Class
FeatureMapResampler
adet/modeling/backbone/bifpn.py:25
↓ 1 callers
Class
GCN
Large Kernel Matters -- https://arxiv.org/abs/1703.02719
adet/layers/gcn.py:52
↓ 1 callers
Class
Hsigmoid
adet/modeling/backbone/vovnet.py:100
↓ 1 callers
Class
IOUMetric
Class to calculate mean-iou using fast_hist method
adet/modeling/MEInst/LME/utils.py:92
↓ 1 callers
Class
MEInstHead
adet/modeling/MEInst/MEInst.py:179
↓ 1 callers
Class
MEInstOutputs
adet/modeling/MEInst/MEInst_outputs.py:57
↓ 1 callers
Class
MaskBranch
adet/modeling/condinst/mask_branch.py:21
↓ 1 callers
Class
MobileNetV2
Should freeze bn
adet/modeling/backbone/mobilenet.py:70
↓ 1 callers
Class
PCAMaskEncoding
To do the mask encoding of PCA. components_: (tensor), shape (n_components, n_features) if agnostic=True
adet/modeling/MEInst/MaskEncoding.py:10
↓ 1 callers
Class
PositionAttention
adet/modeling/abinet/attention.py:19
↓ 1 callers
Class
PositionContentAttention
adet/modeling/abinet/attention.py:81
↓ 1 callers
Class
RNNPredictor
adet/modeling/roi_heads/text_head.py:38
↓ 1 callers
Class
RandomCropWithInstanceAR
Instance-aware cropping.
adet/data/augmentation.py:110
↓ 1 callers
Class
RandomRotationWithProb
adet/data/augmentation.py:140
↓ 1 callers
Class
ResNetLPF
adet/modeling/backbone/resnet_lpf.py:149
↓ 1 callers
Class
Root
adet/modeling/backbone/dla.py:154
↓ 1 callers
Class
SOLOv2InsHead
adet/modeling/solov2/solov2.py:507
↓ 1 callers
Class
SOLOv2MaskHead
adet/modeling/solov2/solov2.py:631
↓ 1 callers
Class
Scale
adet/modeling/batext/batext.py:19
↓ 1 callers
Class
Scale
adet/modeling/fcos/fcos.py:20
↓ 1 callers
Class
Scale
adet/modeling/MEInst/MEInst.py:21
↓ 1 callers
Class
SeqConvs
adet/modeling/roi_heads/text_head.py:16
↓ 1 callers
Class
SingleBiFPN
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 callers
Class
SoftCrossEntropyLoss
adet/utils/comm.py:94
↓ 1 callers
Class
SpellingMutation
adet/data/dataset_language.py:174
↓ 1 callers
Class
TextDataset
adet/data/dataset_language.py:80
↓ 1 callers
Class
TextEvaluator
Evaluate text proposals and recognition.
adet/evaluation/text_evaluation.py:27
↓ 1 callers
Class
TextIterator
adet/data/dataset_language.py:64
↓ 1 callers
Class
Trainer
This is the same Trainer except that we rewrite the `build_train_loader`/`resume_or_load` method.
tools/train_net.py:48
↓ 1 callers
Class
VisualizationDemo
demo/predictor.py:19
↓ 1 callers
Class
VoVNet
adet/modeling/backbone/vovnet.py:211
↓ 1 callers
Class
_OSA_stage
adet/modeling/backbone/vovnet.py:172
↓ 1 callers
Class
eSEModule
adet/modeling/backbone/vovnet.py:109
Class
ABINetIterModel
adet/modeling/abinet/model_abinet_iter.py:11
Class
ABINetModel
adet/modeling/abinet/model_abinet.py:11
Class
BAText
A modified version of FCOS with Bezier regression
adet/modeling/batext/batext.py:29
Class
BasicBlock
adet/modeling/backbone/dla.py:38
Class
BasicBlock
adet/modeling/backbone/resnet_lpf.py:63
Class
BlendMask
Main class for BlendMask architectures (see https://arxiv.org/abd/1901.02446).
adet/modeling/blendmask/blendmask.py:22
Class
Bottleneck
adet/modeling/backbone/dla.py:69
Class
Bottleneck
adet/modeling/backbone/resnet_lpf.py:104
Class
BottleneckX
adet/modeling/backbone/dla.py:110
Class
CondInst
Main class for CondInst architectures (see https://arxiv.org/abs/2003.05664).
adet/modeling/condinst/condinst.py:74
Class
DFConv2d
Deformable convolutional layer with configurable deformable groups, dilations and groups. Code is from: https://github.com/facebookr
adet/layers/deform_conv.py:19
Class
DefROIAlign
adet/layers/def_roi_align.py:60
Class
Downsample1D
adet/modeling/backbone/lpf.py:62
Class
FCOS
Implement FCOS (https://arxiv.org/abs/1904.01355).
adet/modeling/fcos/fcos.py:43
Class
FocalLoss
adet/modeling/solov2/loss.py:91
Class
InvertedResidual
adet/modeling/backbone/mobilenet.py:29
Class
MEInst
Implement MEInst (https://arxiv.org/abs/2003.11712).
adet/modeling/MEInst/MEInst.py:31
Class
Model
adet/modeling/abinet/model.py:5
Class
OneStageDetector
Same as :class:`detectron2.modeling.ProposalNetwork`. Uses "instances" as the return key instead of using "proposal".
adet/modeling/one_stage_detector.py:39
Class
OneStageRCNN
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
Class
PCATextHead
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
Class
PreCalc
adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp:9
Class
ProtoNet
adet/modeling/blendmask/basis_module.py:26
Class
RandomCropWithInstance
Instance-aware cropping.
adet/data/augmentation.py:90
next →
1–100 of 108, ranked by callers