Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/botaoye/OSTrack
/ types & classes
Types & classes
102 in github.com/botaoye/OSTrack
⨍
Functions
788
◇
Types & classes
102
↓ 46 callers
Class
TensorList
Container mainly used for lists of torch tensors. Extends lists with pytorch functionality.
lib/utils/tensor.py:39
↓ 18 callers
Class
SequenceList
List of sequences. Supports the addition operator to concatenate sequence lists.
lib/test/evaluation/data.py:150
↓ 13 callers
Class
Sequence
Class for the sequence in an evaluation.
lib/test/evaluation/data.py:21
↓ 10 callers
Class
TensorDict
Container mainly used for dicts of torch tensors. Extends OrderedDict with pytorch functionality.
lib/utils/tensor.py:7
↓ 5 callers
Class
NestedTensor
lib/utils/misc.py:284
↓ 4 callers
Class
COCO
lib/train/dataset/COCO_tool.py:70
↓ 4 callers
Class
Got10k
GOT-10k dataset. Publication: GOT-10k: A Large High-Diversity Benchmark for Generic Object Tracking in the Wild Lianghua Huang,
lib/train/dataset/got10k.py:14
↓ 4 callers
Class
Tracker
Wraps the tracker for evaluation and running purposes. args: name: Name of tracking method. parameter_name: Name of parameter file
lib/test/evaluation/tracker.py:27
↓ 3 callers
Class
Got10k_lmdb
lib/train/dataset/got10k_lmdb.py:17
↓ 2 callers
Class
Attention
lib/models/layers/attn.py:9
↓ 2 callers
Class
AverageMeter
Computes and stores the average and current value
lib/train/admin/stats.py:19
↓ 2 callers
Class
LTRLoader
Data loader. Combines a dataset and a sampler, and provides single- or multi-process iterators over the dataset. Note: The only differen
lib/train/data/loader.py:125
↓ 2 callers
Class
LTRTrainer
lib/train/trainers/ltr_trainer.py:18
↓ 2 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
lib/utils/misc.py:27
↓ 2 callers
Class
StatValue
lib/train/admin/stats.py:3
↓ 2 callers
Class
VisCostVolume
lib/vis/visdom_cus.py:135
↓ 2 callers
Class
Visdom
lib/vis/visdom_cus.py:435
↓ 1 callers
Class
Attention
lib/models/ostrack/vit.py:39
↓ 1 callers
Class
Block
lib/models/ostrack/vit.py:69
↓ 1 callers
Class
CEBlock
lib/models/layers/attn_blocks.py:77
↓ 1 callers
Class
CenterPredictor
lib/models/layers/head.py:98
↓ 1 callers
Class
Corner_Predictor
Corner Predictor module
lib/models/layers/head.py:24
↓ 1 callers
Class
EnvSettings
lib/test/evaluation/environment.py:5
↓ 1 callers
Class
EnvSettings_ITP
lib/test/evaluation/environment.py:55
↓ 1 callers
Class
FocalLoss
lib/utils/focal_loss.py:8
↓ 1 callers
Class
FrozenBatchNorm2d
BatchNorm2d where the batch statistics and the affine parameters are fixed. Copy-paste from torchvision.misc.ops with added eps before rq
lib/models/layers/frozen_bn.py:4
↓ 1 callers
Class
ImagenetVID
Imagenet VID dataset. Publication: ImageNet Large Scale Visual Recognition Challenge Olga Russakovsky, Jia Deng, Hao Su, Jonatha
lib/train/dataset/imagenetvid.py:17
↓ 1 callers
Class
ImagenetVID_lmdb
Imagenet VID dataset. Publication: ImageNet Large Scale Visual Recognition Challenge Olga Russakovsky, Jia Deng, Hao Su, Jonatha
lib/train/dataset/imagenetvid_lmdb.py:16
↓ 1 callers
Class
Lasot
LaSOT dataset. Publication: LaSOT: A High-quality Benchmark for Large-scale Single Object Tracking Heng Fan, Liting Lin, Fan Yan
lib/train/dataset/lasot.py:14
↓ 1 callers
Class
Lasot_lmdb
lib/train/dataset/lasot_lmdb.py:16
↓ 1 callers
Class
MLP
Very simple multi-layer perceptron (also called FFN)
lib/models/layers/head.py:204
↓ 1 callers
Class
MSCOCOSeq
The COCO dataset. COCO is an image dataset. Thus, we treat each image as a sequence of length 1. Publication: Microsoft COCO: Common Obj
lib/train/dataset/coco_seq.py:11
↓ 1 callers
Class
MSCOCOSeq_lmdb
The COCO dataset. COCO is an image dataset. Thus, we treat each image as a sequence of length 1. Publication: Microsoft COCO: Common Obj
lib/train/dataset/coco_seq_lmdb.py:12
↓ 1 callers
Class
OSTrack
This is the base class for OSTrack
lib/models/ostrack/ostrack.py:18
↓ 1 callers
Class
OSTrackActor
Actor for training OSTrack models
lib/train/actors/ostrack.py:10
↓ 1 callers
Class
PatchEmbed
2D Image to Patch Embedding
lib/models/layers/patch_embed.py:6
↓ 1 callers
Class
Preprocessor
lib/test/tracker/data_utils.py:6
↓ 1 callers
Class
TensorboardWriter
lib/train/admin/tensorboard.py:10
↓ 1 callers
Class
TrackerParams
Class for tracker parameters.
lib/test/utils/params.py:5
↓ 1 callers
Class
TrackingNet
TrackingNet dataset. Publication: TrackingNet: A Large-Scale Dataset and Benchmark for Object Tracking in the Wild. Matthias Mue
lib/train/dataset/tracking_net.py:35
↓ 1 callers
Class
TrackingNet_lmdb
TrackingNet dataset. Publication: TrackingNet: A Large-Scale Dataset and Benchmark for Object Tracking in the Wild. Matthias Mue
lib/train/dataset/tracking_net_lmdb.py:30
↓ 1 callers
Class
VisBBReg
lib/vis/visdom_cus.py:401
↓ 1 callers
Class
VisCostVolumeUI
lib/vis/visdom_cus.py:185
↓ 1 callers
Class
VisFeaturemap
lib/vis/visdom_cus.py:102
↓ 1 callers
Class
VisHeatmap
lib/vis/visdom_cus.py:81
↓ 1 callers
Class
VisImage
lib/vis/visdom_cus.py:69
↓ 1 callers
Class
VisInfoDict
lib/vis/visdom_cus.py:268
↓ 1 callers
Class
VisLinePlot
lib/vis/visdom_cus.py:308
↓ 1 callers
Class
VisResults
tracking/vis_results.py:20
↓ 1 callers
Class
VisText
lib/vis/visdom_cus.py:296
↓ 1 callers
Class
VisTracking
lib/vis/visdom_cus.py:326
↓ 1 callers
Class
VisionTransformer
Vision Transformer A PyTorch impl of : `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale` - https://arxiv.org
lib/models/ostrack/vit.py:94
↓ 1 callers
Class
VisionTransformerCE
Vision Transformer with candidate elimination (CE) module A PyTorch impl of : `An Image is Worth 16x16 Words: Transformers for Image Recogniti
lib/models/ostrack/vit_ce.py:21
↓ 1 callers
Class
WandbWriter
lib/train/data/wandb_logger.py:10
Class
Attention_talking_head
lib/models/layers/attn.py:62
Class
BaseActor
Base class for actor. The actor class handles the passing of the data through the network and calculation the loss
lib/train/actors/base_actor.py:4
Class
BaseBackbone
lib/models/ostrack/base_backbone.py:13
Class
BaseDataset
Base class for all datasets.
lib/test/evaluation/data.py:7
Class
BaseImageDataset
Base class for image datasets
lib/train/dataset/base_image_dataset.py:5
Class
BaseProcessing
Base class for Processing. Processing class is used to process the data returned by a dataset, before passing it through the network. For exampl
lib/train/data/processing.py:14
Class
BaseTracker
Base class for all trackers.
lib/test/tracker/basetracker.py:10
Class
BaseTrainer
Base trainer class. Contains functions for training and saving/loading checkpoints. Trainer classes should inherit from this one and overload the
lib/train/trainers/base_trainer.py:9
Class
BaseVideoDataset
Base class for video datasets
lib/train/dataset/base_video_dataset.py:6
Class
Block
lib/models/layers/attn_blocks.py:106
Class
CenterNetHeatMap
lib/utils/heapmap_utils.py:29
Class
FeatureParams
Class for feature specific parameters
lib/test/utils/params.py:28
Class
GOT10KDataset
GOT-10k dataset. Publication: GOT-10k: A Large High-Diversity Benchmark for Generic Object Tracking in the Wild Lianghua Huang,
lib/test/evaluation/got10kdataset.py:7
Class
ITBDataset
NUS-PRO dataset
lib/test/evaluation/itbdataset.py:7
Class
LBHinge
Loss that uses a 'hinge' on the lower bound. This means that for samples with a label value smaller than the threshold, the loss is zero if the p
lib/utils/focal_loss.py:38
Class
LaSOTDataset
LaSOT test set consisting of 280 videos (see Protocol-II in the LaSOT paper) Publication: LaSOT: A High-quality Benchmark for Large-
lib/test/evaluation/lasotdataset.py:6
Class
LaSOTExtensionSubsetDataset
LaSOT test set consisting of 280 videos (see Protocol-II in the LaSOT paper) Publication: LaSOT: A High-quality Large-scale Single
lib/test/evaluation/lasotextensionsubsetdataset.py:6
Class
LaSOTlmdbDataset
LaSOT test set consisting of 280 videos (see Protocol-II in the LaSOT paper) Publication: LaSOT: A High-quality Benchmark for Large-
lib/test/evaluation/lasot_lmdbdataset.py:7
Class
MSCOCO
The COCO object detection dataset. Publication: Microsoft COCO: Common Objects in Context. Tsung-Yi Lin, Michael Maire, Serge J.
lib/train/dataset/coco.py:11
Class
MetricLogger
lib/utils/misc.py:159
Class
MultiGPU
lib/train/admin/multigpu.py:9
Class
NFSDataset
NFS dataset. Publication: Need for Speed: A Benchmark for Higher Frame Rate Object Tracking H. Kiani Galoogahi, A. Fagg, C. Huang
lib/test/evaluation/nfsdataset.py:6
Class
Normalize
Normalize image
lib/train/data/transforms.py:243
Class
OSTrack
lib/test/tracker/ostrack.py:19
Class
OTBDataset
OTB-2015 dataset Publication: Object Tracking Benchmark Wu, Yi, Jongwoo Lim, and Ming-hsuan Yan TPAMI, 2015 http:
lib/test/evaluation/otbdataset.py:6
Class
PreprocessorX
lib/test/tracker/data_utils.py:20
Class
PreprocessorX_onnx
lib/test/tracker/data_utils.py:34
Class
RandomHorizontalFlip
Horizontally flip image randomly with a probability p.
lib/train/data/transforms.py:284
Class
RandomHorizontalFlip_Norm
Horizontally flip image randomly with a probability p. The difference is that the coord is normalized to [0,1]
lib/train/data/transforms.py:322
Class
RelativePosition2DEncoder
lib/models/layers/rpe.py:91
Class
STARKProcessing
The processing class used for training LittleBoy. The images are processed in the following way. First, the target bounding box is jittered by ad
lib/train/data/processing.py:38
Class
Settings
Training settings, e.g. the paths to datasets and networks.
lib/train/admin/settings.py:4
Class
TC128CEDataset
TC-128 Dataset (78 newly added sequences) modified from the implementation in got10k-toolkit (https://github.com/got-10k/toolkit)
lib/test/evaluation/tc128cedataset.py:8
Class
TC128Dataset
TC-128 Dataset modified from the implementation in got10k-toolkit (https://github.com/got-10k/toolkit)
lib/test/evaluation/tc128dataset.py:8
Class
TNL2kDataset
TNL2k test set
lib/test/evaluation/tnl2kdataset.py:12
Class
ToBGR
Converts image to BGR
lib/train/data/transforms.py:275
Class
ToGrayscale
Converts image to grayscale with probability
lib/train/data/transforms.py:255
Class
ToTensor
Convert to a Tensor
lib/train/data/transforms.py:181
Class
ToTensorAndJitter
Convert to a Tensor and jitter brightness
lib/train/data/transforms.py:209
Class
TrackingNetDataset
TrackingNet test set. Publication: TrackingNet: A Large-Scale Dataset and Benchmark for Object Tracking in the Wild. Matthias Mu
lib/test/evaluation/trackingnetdataset.py:7
Class
TrackingSampler
Class responsible for sampling frames from training sequences to form batches. The sampling is done in the following ways. First a dataset is s
lib/train/data/sampler.py:11
Class
Transform
A set of transformations, used for e.g. data augmentation. Args of constructor: transforms: An arbitrary number of transformations, derive
lib/train/data/transforms.py:10
Class
TransformBase
Base class for transformation objects. See the Transform class for details.
lib/train/data/transforms.py:86
Class
UAVDataset
UAV123 dataset. Publication: A Benchmark and Simulator for UAV Tracking. Matthias Mueller, Neil Smith and Bernard Ghanem
lib/test/evaluation/uavdataset.py:6
Class
VOTDataset
VOT2018 dataset Publication: The sixth Visual Object Tracking VOT2018 challenge results. Matej Kristan, Ales Leonardis, Jiri
lib/test/evaluation/votdataset.py:9
Class
VisBase
lib/vis/visdom_cus.py:38
next →
1–100 of 102, ranked by callers