Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Kitware/COAT
/ types & classes
Types & classes
37 in github.com/Kitware/COAT
⨍
Functions
154
◇
Types & classes
37
↓ 3 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
utils/utils.py:24
↓ 3 callers
Class
TransformerHead
models/transformer.py:19
↓ 2 callers
Class
KMNode
utils/km.py:10
↓ 2 callers
Class
SoftmaxLoss
loss/softmax_loss.py:9
↓ 2 callers
Class
exchange_patch
utils/mask.py:135
↓ 1 callers
Class
Attention
Compute 'Scaled Dot Product Attention
models/transformer.py:224
↓ 1 callers
Class
BBoxRegressor
Bounding box regression layer.
models/coat.py:656
↓ 1 callers
Class
Backbone
models/resnet.py:10
↓ 1 callers
Class
COAT
models/coat.py:23
↓ 1 callers
Class
CUHKSYSU
datasets/cuhk_sysu.py:10
↓ 1 callers
Class
CascadedROIHeads
https://github.com/pytorch/vision/blob/master/torchvision/models/detection/roi_heads.py
models/coat.py:185
↓ 1 callers
Class
Compose
utils/transforms.py:33
↓ 1 callers
Class
Cutout
Randomly mask out one or more patches from an image. https://github.com/uoguelph-mlrg/Cutout/blob/master/util/cutout.py Args: n_holes
utils/transforms.py:56
↓ 1 callers
Class
FeedForward
models/transformer.py:288
↓ 1 callers
Class
KuhnMunkres
utils/km.py:18
↓ 1 callers
Class
MetricLogger
utils/utils.py:88
↓ 1 callers
Class
MultiHeadedAttention
Take in model size and number of heads.
models/transformer.py:258
↓ 1 callers
Class
NormAwareEmbedding
Implements the Norm-Aware Embedding proposed in Chen, Di, et al. "Norm-aware embedding for efficient person search." CVPR 2020.
models/coat.py:584
↓ 1 callers
Class
OIMLoss
loss/oim.py:41
↓ 1 callers
Class
PRW
datasets/prw.py:14
↓ 1 callers
Class
RandomErasing
https://github.com/zhunzhong07/CamStyle/blob/master/reid/utils/data/transforms.py
utils/transforms.py:94
↓ 1 callers
Class
RandomHorizontalFlip
utils/transforms.py:43
↓ 1 callers
Class
Res5Head
models/resnet.py:33
↓ 1 callers
Class
ToTensor
utils/transforms.py:127
↓ 1 callers
Class
Transformer
models/transformer.py:141
↓ 1 callers
Class
TransformerBlock
Transformer = MultiHead_Attention + Feed_Forward with sublayer connection
models/transformer.py:203
↓ 1 callers
Class
Transformers
models/transformer.py:69
↓ 1 callers
Class
Vec2Patch
models/transformer.py:241
↓ 1 callers
Class
cutout_patch
utils/mask.py:221
↓ 1 callers
Class
erase_patch
utils/mask.py:246
↓ 1 callers
Class
exchange_token
utils/mask.py:8
↓ 1 callers
Class
get_mask_box
utils/mask.py:54
↓ 1 callers
Class
jigsaw_patch
utils/mask.py:298
↓ 1 callers
Class
jigsaw_token
utils/mask.py:21
↓ 1 callers
Class
mixup_patch
utils/mask.py:272
Class
BaseDataset
Base class of person search dataset.
datasets/base.py:8
Class
OIM
loss/oim.py:9