Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/whai362/PVT
/ types & classes
Types & classes
58 in github.com/whai362/PVT
⨍
Functions
229
◇
Types & classes
58
↓ 7 callers
Class
PyramidVisionTransformerV2
classification/pvt_v2.py:215
↓ 5 callers
Class
PyramidVisionTransformer
classification/pvt.py:130
↓ 2 callers
Class
INatDataset
classification/datasets.py:14
↓ 2 callers
Class
ImageNet
classification/mcloader/imagenet.py:4
↓ 2 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
classification/utils.py:18
↓ 1 callers
Class
Attention
detection/pvt.py:33
↓ 1 callers
Class
Attention
detection/pvt_v2.py:57
↓ 1 callers
Class
Attention
classification/pvt.py:34
↓ 1 callers
Class
Attention
classification/pvt_v2.py:54
↓ 1 callers
Class
Attention
segmentation/pvt.py:33
↓ 1 callers
Class
Block
detection/pvt.py:78
↓ 1 callers
Class
Block
detection/pvt_v2.py:132
↓ 1 callers
Class
Block
classification/pvt.py:79
↓ 1 callers
Class
Block
classification/pvt_v2.py:129
↓ 1 callers
Class
Block
segmentation/pvt.py:78
↓ 1 callers
Class
ClassificationDataset
Dataset for classification.
classification/mcloader/classification.py:6
↓ 1 callers
Class
DWConv
detection/pvt_v2.py:316
↓ 1 callers
Class
DWConv
classification/pvt_v2.py:304
↓ 1 callers
Class
DistillationLoss
This module wraps a standard criterion and adds an extra knowledge distillation loss by taking a teacher model prediction and using it as add
classification/losses.py:10
↓ 1 callers
Class
McLoader
classification/mcloader/mcloader.py:14
↓ 1 callers
Class
Mlp
detection/pvt.py:14
↓ 1 callers
Class
Mlp
detection/pvt_v2.py:15
↓ 1 callers
Class
Mlp
classification/pvt.py:15
↓ 1 callers
Class
Mlp
classification/pvt_v2.py:12
↓ 1 callers
Class
Mlp
segmentation/pvt.py:14
↓ 1 callers
Class
OverlapPatchEmbed
Image to Patch Embedding
detection/pvt_v2.py:172
↓ 1 callers
Class
OverlapPatchEmbed
Image to Patch Embedding
classification/pvt_v2.py:169
↓ 1 callers
Class
PatchEmbed
Image to Patch Embedding
detection/pvt.py:101
↓ 1 callers
Class
PatchEmbed
Image to Patch Embedding
classification/pvt.py:102
↓ 1 callers
Class
PatchEmbed
Image to Patch Embedding
segmentation/pvt.py:101
↓ 1 callers
Class
RASampler
Sampler that restricts data loading to a subset of the dataset for distributed, with repeated augmentation. It ensures that different each aug
classification/samplers.py:8
↓ 1 callers
Class
ResultVisualizer
Display and save evaluation results. Args: show (bool): Whether to show the image. Default: True wait_time (float): Value of wait
detection/analyze_results.py:48
↓ 1 callers
Class
Trainer
classification/run_with_submitit.py:49
Class
AlignResize
Resize images & seg. Align
segmentation/align_resize.py:10
Class
DataPrefetcher
classification/mcloader/data_prefetcher.py:4
Class
DistOptimizerHook
Optimizer hook for distributed training.
detection/mmcv_custom/runner/optimizer.py:9
Class
EpochBasedRunnerAmp
Epoch-based Runner with AMP support. This runner train models epoch by epoch.
detection/mmcv_custom/runner/epoch_based_runner.py:20
Class
ImageList
classification/mcloader/image_list.py:7
Class
MetricLogger
classification/utils.py:80
Class
PyramidVisionTransformer
detection/pvt.py:129
Class
PyramidVisionTransformer
segmentation/pvt.py:129
Class
PyramidVisionTransformerV2
detection/pvt_v2.py:217
Class
pvt_large
detection/pvt.py:272
Class
pvt_large
segmentation/pvt.py:263
Class
pvt_medium
detection/pvt.py:263
Class
pvt_medium
segmentation/pvt.py:254
Class
pvt_small
detection/pvt.py:245
Class
pvt_small
segmentation/pvt.py:245
Class
pvt_small_f4
detection/pvt.py:254
Class
pvt_tiny
detection/pvt.py:236
Class
pvt_tiny
segmentation/pvt.py:236
Class
pvt_v2_b0
detection/pvt_v2.py:342
Class
pvt_v2_b1
detection/pvt_v2.py:351
Class
pvt_v2_b2
detection/pvt_v2.py:360
Class
pvt_v2_b2_li
detection/pvt_v2.py:369
Class
pvt_v2_b3
detection/pvt_v2.py:378
Class
pvt_v2_b4
detection/pvt_v2.py:387
Class
pvt_v2_b5
detection/pvt_v2.py:396