Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/BoPang1996/TubeTK
/ functions
Functions
287 in github.com/BoPang1996/TubeTK
⨍
Functions
287
◇
Types & classes
60
↓ 1 callers
Function
main
(train_arg, model_arg)
main.py:151
↓ 1 callers
Function
main
(test_arg, model_arg)
evaluate.py:105
↓ 1 callers
Function
main
(test_arg, model_arg)
demo.py:300
↓ 1 callers
Method
mirror
(self, images, tubes)
dataset/augmentation.py:415
↓ 1 callers
Function
multiclass_nms
NMS for multi-class tubes. Args: multi_tubes (Tensor): shape (n, #class*4) or (n, 4) multi_scores (Tensor): shape (n, 1+#class)
post_processing/tube_nms.py:6
↓ 1 callers
Function
new_tracks
(tracks, tubes)
post_processing/tube_iou_matching_old.py:264
↓ 1 callers
Function
new_tracks
(tracks, tubes)
post_processing/tube_iou_matching_super_old.py:247
↓ 1 callers
Function
one_hot_embedding
Embedding labels to one-hot form. Args: labels: (LongTensor) class labels, sized [N,]. num_classes: (int) number of classes.
network/focal_loss.py:6
↓ 1 callers
Function
parse_args
()
launch.py:7
↓ 1 callers
Function
preprocessResult
(resFile, seqName, dataDir=None, force=True, minvis=0.0)
Metrics.py:101
↓ 1 callers
Function
resnet101
Constructs a ResNet-101 model.
network/resnet.py:183
↓ 1 callers
Function
resnet50
Constructs a ResNet-50 model.
network/resnet.py:170
↓ 1 callers
Function
resnext101
Constructs a ResNet-101 model.
network/resnet.py:196
↓ 1 callers
Method
run
(self)
pre_processing/get_tubes_MOT17.py:180
↓ 1 callers
Method
run
(self)
pre_processing/get_tubes_jta.py:200
↓ 1 callers
Function
run_one_iter
(model, optimizer, data, scheduler, test)
main.py:48
↓ 1 callers
Function
synchronize
Helper function to synchronize (barrier) among all processes when using distributed training
evaluate.py:20
↓ 1 callers
Function
synchronize
Helper function to synchronize (barrier) among all processes when using distributed training
demo.py:215
↓ 1 callers
Function
track_complete
(track, gap_threshold=8)
post_processing/tube_iou_matching.py:225
↓ 1 callers
Function
track_tube_iou
(track_boxes, tube_boxes)
post_processing/tube_iou_matching.py:63
↓ 1 callers
Function
track_tube_overlaps
(bboxes1, bboxes2)
post_processing/tube_iou_matching.py:67
↓ 1 callers
Function
trackfilt
(track, l=16)
post_processing/tube_iou_matching.py:203
↓ 1 callers
Function
train
(model, optimizer, data_loader, scheduler, writer, max_acc=0, step_start=0)
main.py:94
↓ 1 callers
Method
train
(self, mode=True)
network/resnet.py:160
↓ 1 callers
Function
tube_giou
(pred_tubes, target_tubes)
network/utils.py:72
↓ 1 callers
Function
tube_iou
(pred_tubes, target_tubes)
network/utils.py:123
↓ 1 callers
Function
tubes2bbox
(tubes, colors, tube_len, stride)
Visualization/Vis_Input.py:38
↓ 1 callers
Function
tubes2bbox_out
(tubes, colors, tube_len, stride)
Visualization/Vis_Input.py:76
↓ 1 callers
Method
tubetk_target
Args: points: 5 * [n_point, 3] gt_tubes_list: b, n_tube, 15 gt_labels_list: b, n_tube, 1
network/track_head.py:319
↓ 1 callers
Function
update_tracks
(tracks, tube, arg)
post_processing/tube_iou_matching_old.py:165
↓ 1 callers
Function
update_tracks
(tracks, tube, arg)
post_processing/tube_iou_matching_super_old.py:150
↓ 1 callers
Function
update_tracks_fast
(tracks, tube, arg)
post_processing/tube_iou_matching.py:101
↓ 1 callers
Function
vis_res
(args)
Visualization/Vis_Res.py:70
Function
PYBIND11_MODULE
post_processing/nms/src/nms_cuda.cpp:15
Function
PYBIND11_MODULE
post_processing/nms/src/nms_cpu.cpp:69
Method
__call__
(self, images, img_meta, tubes, labels, start_frame)
Visualization/Vis_Input.py:18
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:49
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:57
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:65
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:76
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:86
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:96
Method
__call__
(self, images, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:208
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:232
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:245
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:261
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:275
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:295
Method
__call__
(self, image, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:308
Method
__call__
(self, images, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:363
Method
__call__
(self, images, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:397
Method
__call__
(self, images, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:424
Method
__call__
Args: image (Tensor): image tensor to be transformed Return: a tensor with channels swapped according to swap
dataset/augmentation.py:444
Method
__call__
(self, images, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:472
Method
__call__
(self, images, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:487
Method
__call__
(self, images, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:500
Method
__call__
(self, images, img_meta, tubes, labels, start_frame)
dataset/augmentation.py:536
Method
__del__
(self)
post_processing/tube_iou_matching_old.py:44
Method
__del__
(self)
post_processing/tube_iou_matching_super_old.py:41
Method
__getitem__
(self, item)
demo.py:129
Method
__getitem__
(self, item)
demo.py:167
Method
__getitem__
(self, item)
dataset/jta.py:30
Method
__getitem__
(self, item)
dataset/mot17.py:31
Method
__getitem__
(self, item)
dataset/mot17.py:73
Method
__getitem__
(self, item)
dataset/mot17jta.py:42
Method
__getitem__
(self, item)
dataset/Parsers/JTA.py:148
Method
__getitem__
(self, item)
dataset/Parsers/MOT17.py:139
Method
__getitem__
(self, item)
post_processing/tube_iou_matching_old.py:116
Method
__getitem__
(self, item)
post_processing/tube_iou_matching_super_old.py:103
Method
__init__
(self, video, forward_frames=4, frame_stride=1, min_vis=-0.
demo.py:23
Method
__init__
(self, data_root, forward_frames=4, frame_stride=1, min_vis
demo.py:108
Method
__init__
(self, data_root, arg, transform=SSJAugmentation,
demo.py:153
Method
__init__
(self, batch_size, num_workers, input_path,
demo.py:186
Method
__init__
(self, frame, detail=True, path='', verbose=False, device=0)
utils/mem_track.py:19
Method
__init__
(self)
utils/util.py:4
Method
__init__
(self, size=896, mean=(104, 117, 123))
Visualization/Vis_Input.py:9
Method
__init__
(self, jta_root, epoch, arg, transform=SSJ
dataset/jta.py:14
Method
__init__
(self, transforms)
dataset/augmentation.py:46
Method
__init__
(self, mean)
dataset/augmentation.py:62
Method
__init__
(self, mean)
dataset/augmentation.py:73
Method
__init__
(self, size)
dataset/augmentation.py:106
Method
__init__
(self, lower=0.7, upper=1.5)
dataset/augmentation.py:226
Method
__init__
(self, delta=18.0)
dataset/augmentation.py:241
Method
__init__
(self)
dataset/augmentation.py:256
Method
__init__
(self, current='BGR', transform='HSV')
dataset/augmentation.py:271
Method
__init__
(self, lower=0.7, upper=1.5)
dataset/augmentation.py:288
Method
__init__
(self, delta=32)
dataset/augmentation.py:303
Method
__init__
(self, mean=(104, 117, 123))
dataset/augmentation.py:386
Method
__init__
(self, swaps)
dataset/augmentation.py:441
Method
__init__
(self)
dataset/augmentation.py:460
Method
__init__
(self, size=896, mean=(104, 117, 123), type='train')
dataset/augmentation.py:510
Method
__init__
(self, batch_size, num_workers, input_path,
dataset/dataLoader.py:13
Method
__init__
(self, mot_root, epoch, arg, transform=SSJ
dataset/mot17.py:15
Method
__init__
(self, mot_root, epoch, type, test_seq,
dataset/mot17.py:55
Method
__init__
(self, mot17_root, mot15_root, jta_root, e
dataset/mot17jta.py:15
Method
__init__
(self, folder, forward_frames=8, frame_stride=1, min_vis=-0
dataset/Parsers/JTA.py:13
Method
__init__
(self, jta_root, type='train', forward_frames=4, frame_stri
dataset/Parsers/JTA.py:122
Method
__init__
(self, folder, forward_frames=8, frame_stride=1, min_vis=-0
dataset/Parsers/MOT17.py:13
Method
__init__
(self, mot_root, type='train', test_seq=None, forward_frame
dataset/Parsers/MOT17.py:104
Method
__init__
(self, box, frame_id, next_fram_id=-1)
dataset/Parsers/structures.py:2
← previous
next →
101–200 of 287, ranked by callers