Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Lu-Feng/DHE-VPR
/ types & classes
Types & classes
25 in github.com/Lu-Feng/DHE-VPR
⨍
Functions
131
◇
Types & classes
25
↓ 2 callers
Class
DropPath
Obtained from: github.com:rwightman/pytorch-image-models Drop paths (Stochastic Depth) per sample (when applied in main path of residual blo
model/cct/stochastic_depth.py:28
↓ 1 callers
Class
Attention
Obtained from timm: github.com:rwightman/pytorch-image-models
model/cct/transformers.py:7
↓ 1 callers
Class
CCT
model/cct/cct.py:32
↓ 1 callers
Class
Flatten
network.py:39
↓ 1 callers
Class
GeM
network.py:26
↓ 1 callers
Class
L2Norm
network.py:46
↓ 1 callers
Class
MaskedAttention
model/cct/transformers.py:38
↓ 1 callers
Class
MaskedTransformerEncoderLayer
Inspired by torch.nn.TransformerEncoderLayer and timm.
model/cct/transformers.py:103
↓ 1 callers
Class
RAMEfficient2DMatrix
This class behaves similarly to a numpy.ndarray initialized with np.zeros(), but is implemented to save RAM when the rows within the 2D array
datasets_ws.py:359
↓ 1 callers
Class
RAMEfficient4DMatrix
This class behaves similarly to a numpy.ndarray initialized with np.zeros(), but is implemented to save RAM when the rows within the 3D array
datasets_ws.py:380
↓ 1 callers
Class
Tokenizer
model/cct/tokenizer.py:6
↓ 1 callers
Class
TransformerClassifier
model/cct/transformers.py:133
↓ 1 callers
Class
TransformerEncoderLayer
Inspired by torch.nn.TransformerEncoderLayer and timm.
model/cct/transformers.py:73
Class
BaseDataset
Dataset with images from database and queries, used for inference (testing and building cache).
datasets_ws.py:63
Class
DatasetQP
dataset_qp.py:8
Class
Embedder
model/cct/embedder.py:4
Class
FeaturesExtractor
The FeaturesExtractor is composed of two parts: the backbone encoder and the pooling/aggregation layer. The pooling/aggregation layer is used
network.py:53
Class
GeolocDataset
dataset_geoloc.py:12
Class
HomographyRegression
network.py:86
Class
InfiniteDataLoader
commons.py:38
Class
MaskedTransformerClassifier
model/cct/transformers.py:231
Class
Network
Overview of the network: name input output FeaturesExtractor: (2B x 3 x H x W)
network.py:119
Class
PCADataset
datasets_ws.py:51
Class
TextTokenizer
model/cct/tokenizer.py:52
Class
TripletsDataset
Dataset used for training, it is used to compute the triplets with TripletsDataset.compute_triplets() with various mining methods. If is_infe
datasets_ws.py:143