Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/aim-uofa/AdelaiDet
/ types & classes
Types & classes
88 in github.com/aim-uofa/AdelaiDet
⨍
Functions
505
◇
Types & classes
88
↓ 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
↓ 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
↓ 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
↓ 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
Blender
adet/modeling/blendmask/blender.py:12
↓ 2 callers
Class
DatasetMapperWithBasis
This caller enables the default Detectron2 mapper to read an additional basis semantic label
adet/data/dataset_mapper.py:52
↓ 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
_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
BATextOutputs
adet/modeling/batext/batext_outputs.py:126
↓ 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
CRNN
adet/modeling/roi_heads/attn_predictor.py:29
↓ 1 callers
Class
DynamicMaskHead
adet/modeling/condinst/dynamic_mask_head.py:94
↓ 1 callers
Class
FCOS
Implement FCOS (https://arxiv.org/abs/1904.01355).
adet/modeling/fcos/fcos.py:43
↓ 1 callers
Class
FCOSHead
adet/modeling/batext/batext.py:165
↓ 1 callers
Class
FCOSHead
adet/modeling/fcos/fcos.py:125
↓ 1 callers
Class
FCOSOutputs
adet/modeling/fcos/fcos_outputs.py:52
↓ 1 callers
Class
FCPoseDatasetMapper
This caller enables the default Detectron2 mapper to read an additional basis semantic label
adet/data/fcpose_dataset_mapper.py:31
↓ 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
HeatmapGenerator
adet/data/detection_utils.py:109
↓ 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
MaskHead
adet/modeling/roi_heads/text_head.py:70
↓ 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
RNNPredictor
adet/modeling/roi_heads/text_head.py:38
↓ 1 callers
Class
RandomCropWithInstance
Instance-aware cropping.
adet/data/augmentation.py:92
↓ 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
TextEvaluator
Evaluate text proposals and recognition.
adet/evaluation/text_evaluation_all.py:28
↓ 1 callers
Class
TextVisualizer
adet/utils/visualizer.py:7
↓ 1 callers
Class
TopPooler
ROIPooler with option to assign level by max length. Used by top modules.
adet/modeling/poolers.py:83
↓ 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:47
↓ 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
basis_module
adet/modeling/fcpose/basis_module.py:19
↓ 1 callers
Class
eSEModule
adet/modeling/backbone/vovnet.py:109
↓ 1 callers
Class
fcpose_head_module
adet/modeling/fcpose/fcpose_head.py:93
↓ 1 callers
Class
folder
adet/modeling/fcpose/utils.py:44
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
FCPose
adet/modeling/fcpose/fcpose_framework.py:20
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
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
ProtoNet
adet/modeling/blendmask/basis_module.py:26
Class
SOLOv2
SOLOv2 model. Creates FPN backbone, instance branch for kernels and categories prediction, mask branch for unified mask features. Calcula
adet/modeling/solov2/solov2.py:26
Class
TextHead
TextHead performs text region alignment and recognition. It is a simplified ROIHeads, only ground truth RoIs are used during trainin
adet/modeling/roi_heads/text_head.py:127
Class
_BezierAlign
adet/layers/bezier_align.py:10
Class
_DefROIAlign
adet/layers/def_roi_align.py:10
Class
_NewEmptyTensorOp
adet/layers/deform_conv.py:7
Class
_PredictWorker
demo/predictor.py:170
Class
_StopToken
demo/predictor.py:167