MCPcopy Create free account

hub / github.com/HYUNJS/SGT / functions

Functions586 in github.com/HYUNJS/SGT

↓ 2 callersMethodnext_id
()
projects/FairMOT/fairmot/tracker/basetrack.py:35
↓ 2 callersFunctionone_imshow
(img, format, show)
projects/Datasets/MOT/vis/simple_visualizer.py:14
↓ 2 callersMethodoverlay_instances
Args: boxes (Boxes, RotatedBoxes or ndarray): either a :class:`Boxes`, or an Nx4 numpy array of XYXY_ABS format f
projects/Datasets/MOT/vis/simple_visualizer.py:132
↓ 2 callersMethodpreprocess_image
Normalize, pad and batch the input images.
projects/CenterNet/centernet/modeling/meta_arch/centernet.py:103
↓ 2 callersFunctionpseudo_nms
r""" apply max pooling to get the same effect of nms Args: fmap(Tensor): output tensor of previous step pool_size(int):
projects/SGT/sgt/utils.py:95
↓ 2 callersMethodpseudo_nms
r""" apply max pooling to get the same effect of nms Args: fmap(Tensor): output tensor of previous step pool_
projects/CenterNet/centernet/generator/centernet_decode.py:163
↓ 2 callersFunctionremove_duplicate_stracks
(stracksa, stracksb)
projects/FairMOT/fairmot/tracker/multitracker.py:213
↓ 2 callersMethodresume_or_load
(self, resume=True)
projects/FairMOT/fairmot/trainer.py:55
↓ 2 callersMethodtest
Args: cfg (CfgNode): model (nn.Module): evaluators (list[DatasetEvaluator] or None): if None, will ca
projects/SGT/sgt/trainer.py:101
↓ 2 callersMethodto_xyah
(self)
projects/FairMOT/fairmot/tracker/multitracker.py:167
↓ 2 callersMethodtopk_score
get top K point in score map
projects/CenterNet/centernet/generator/centernet_decode.py:177
↓ 2 callersMethodtrainval_split
(self)
projects/Datasets/MOT/gen_trainval.py:18
↓ 1 callersMethod__init__
(self, cfg)
projects/SGT/sgt/meta_arch/graph_heads.py:11
↓ 1 callersMethod__init__
(self, eval_period, eval_function)
projects/EpochTrainer/epoch_trainer/epoch_hooks.py:36
↓ 1 callersMethod__init__
(self, in_channel, inter_channels, out_channel, bias_fill=False, bias_value=0, aspp_on=False)
projects/CenterNet/centernet/layers/centernet_head.py:9
↓ 1 callersMethod__init__
(self, in_channels, atrous_rates, out_channels)
projects/CenterNet/centernet/layers/aspp.py:18
↓ 1 callersMethod_align_image
(self, images, batched_inputs)
projects/CenterNet/centernet/modeling/meta_arch/centernet.py:119
↓ 1 callersMethod_convert_ids
(self, ids)
projects/Datasets/MOT/vis/simple_visualizer.py:439
↓ 1 callersMethod_deconv_init
(self)
projects/CenterNet/centernet/layers/centernet_deconv.py:54
↓ 1 callersMethod_do_eval
(self)
projects/EpochTrainer/epoch_trainer/epoch_hooks.py:39
↓ 1 callersMethod_eval_predictions
(self, predictions)
projects/Datasets/MOT/evaluation/mot_evaluation.py:295
↓ 1 callersMethod_get_low_dets
(self, output)
projects/FairMOT/fairmot/meta_arch/byte_tracker.py:70
↓ 1 callersMethod_load_file
(self, filename)
projects/CenterNet/centernet/checkpoint/centernet_checkpoint.py:43
↓ 1 callersMethod_load_model
Load weights from a checkpoint. ckpt trained on this detectron2 framework: checkpoint_state_dict['model'] Offici
projects/SGT/sgt/checkpointer.py:112
↓ 1 callersMethod_postprocess
(self, results, img_info)
projects/SGT/sgt/meta_arch/sgt.py:86
↓ 1 callersMethod_postprocess
(self, output)
projects/FairMOT/fairmot/meta_arch/jde_tracker.py:65
↓ 1 callersMethod_postprocess
(self, output)
projects/FairMOT/fairmot/meta_arch/byte_tracker.py:60
↓ 1 callersMethod_postprocess
(self, results, img_info, batched_inputs=None)
projects/CenterNet/centernet/modeling/meta_arch/centernet.py:148
↓ 1 callersFunction_reg_loss
L1 regression loss Arguments: regr (batch x max_objects x dim) gt_regr (batch x max_objects x dim) mask (batch x max_objects)
projects/CenterNet/centernet/losses.py:95
↓ 1 callersFunction_sigmoid
(x)
projects/CenterNet/centernet/centernet_loss.py:9
↓ 1 callersMethod_write_metrics
(self, loss_dict, data_time, prefix="train")
projects/SGT/sgt/trainer.py:166
↓ 1 callersMethod_write_metrics
Args: loss_dict (dict): dict of scalar losses data_time (float): time taken by the dataloader iteration
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:493
↓ 1 callersMethod_write_metrics
(self, loss_dict, data_time, prefix="train")
projects/FairMOT/fairmot/trainer.py:134
↓ 1 callersMethod_write_metrics
(self, loss_dict, data_time, prefix="train")
projects/CenterNet/centernet/trainer.py:75
↓ 1 callersFunctionadd_fairmot_config
(cfg: CN)
projects/FairMOT/fairmot/config.py:5
↓ 1 callersFunctionadd_random_affine_config
Add config for MOT dataset.
projects/Datasets/Transforms/config.py:4
↓ 1 callersFunctionadd_sgt_config
(cfg: CN)
projects/SGT/sgt/config.py:5
↓ 1 callersMethodadd_tracklets
(self, t1_info, not_found_mask_t1)
projects/SGT/sgt/tracker/tracklet_manager.py:38
↓ 1 callersFunctionadjust_format_v1
(filename)
projects/Datasets/Hieve/submit_hieve.py:12
↓ 1 callersMethodafter_step
(self)
projects/EpochTrainer/epoch_trainer/epoch_hooks.py:62
↓ 1 callersMethodafter_train
(self)
projects/EpochTrainer/epoch_trainer/epoch_hooks.py:68
↓ 1 callersMethodassign_gt_ids
(self, targets, raw_detections, decoded_detections)
projects/SGT/sgt/meta_arch/graphtracker_centernet.py:280
↓ 1 callersMethodassign_pred_ids
(self, topk_scores)
projects/SGT/sgt/meta_arch/graphtracker.py:379
↓ 1 callersMethodauto_scale_workers
When the config is defined for certain number of workers (according to ``cfg.SOLVER.REFERENCE_WORLD_SIZE``) that's different from the
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:423
↓ 1 callersMethodbefore_train
(self)
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:229
↓ 1 callersFunctionbuild_COCO_transform_gen
(cfg, is_train)
projects/CenterNet/centernet/dataset_mapper.py:12
↓ 1 callersFunctionbuild_MOT_transform_gen
(cfg, is_train)
projects/CenterNet/centernet/dataset_mapper.py:27
↓ 1 callersMethodbuild_MPNet
(self, node_model_param, edge_model_param)
projects/SGT/sgt/meta_arch/graph_net.py:36
↓ 1 callersFunctionbuild_centernet_backbone
(cfg)
projects/CenterNet/centernet/build.py:10
↓ 1 callersFunctionbuild_centernet_head
(cfg, in_channels)
projects/CenterNet/centernet/build.py:27
↓ 1 callersFunctionbuild_centernet_upsample_layer
(cfg)
projects/CenterNet/centernet/build.py:17
↓ 1 callersMethodbuild_checkpointer
(cls, model, output_dir, optimizer, scheduler)
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:322
↓ 1 callersFunctionbuild_detector
Build the whole model architecture, defined by ``cfg.MODEL.META_ARCHITECTURE``. Note that it does not load any weights from ``cfg``.
projects/SGT/sgt/build.py:6
↓ 1 callersFunctionbuild_detector
Build the whole model architecture, defined by ``cfg.MODEL.META_ARCHITECTURE``. Note that it does not load any weights from ``cfg``.
projects/FairMOT/fairmot/build.py:6
↓ 1 callersMethodbuild_evaluator
(cls, cfg, dataset_name, output_folder=None)
projects/SGT/sgt/trainer.py:81
↓ 1 callersMethodbuild_evaluator
Returns: DatasetEvaluator or None It is not implemented by default.
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:354
↓ 1 callersMethodbuild_evaluator
(cls, cfg, dataset_name, output_folder=None)
projects/FairMOT/fairmot/trainer.py:50
↓ 1 callersMethodbuild_hooks
Build a list of default hooks, including timing, evaluation, checkpointing, lr scheduling, precise BN, writing events. Retur
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:119
↓ 1 callersMethodbuild_idx_per_rank
(self)
projects/Datasets/MOT/sequence_distributed_sampler.py:29
↓ 1 callersMethodbuild_idx_per_seq
(self, seq_name_list)
projects/Datasets/MOT/sequence_distributed_sampler.py:21
↓ 1 callersMethodbuild_lr_scheduler
It now calls :func:`detectron2.solver.build_lr_scheduler`. Overwrite it if you'd like a different scheduler.
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:314
↓ 1 callersFunctionbuild_optimizer
Build an optimizer from config.
projects/EpochTrainer/epoch_trainer/build.py:9
↓ 1 callersMethodbuild_optimizer
(cls, cfg, model)
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:264
↓ 1 callersFunctionbuild_sparse_graph
Suppose the number of previous boxes (t1) is dynamic so the boxes in the batches are listed. In contrast, the number of current boxes (t2)
projects/SGT/sgt/meta_arch/graph_build.py:8
↓ 1 callersMethodbuild_test_loader
(cls, cfg, dataset_name)
projects/SGT/sgt/trainer.py:76
↓ 1 callersMethodbuild_test_loader
Returns: iterable It now calls :func:`detectron2.data.build_detection_test_loader`. Overwrite it if you'd like a
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:343
↓ 1 callersMethodbuild_test_loader
(cls, cfg, dataset_name)
projects/FairMOT/fairmot/trainer.py:45
↓ 1 callersFunctionbuild_tracker
Build the whole model architecture, defined by ``cfg.MODEL.META_ARCHITECTURE``. Note that it does not load any weights from ``cfg``.
projects/SGT/sgt/build.py:17
↓ 1 callersFunctionbuild_tracker
Build the whole model architecture, defined by ``cfg.MODEL.META_ARCHITECTURE``. Note that it does not load any weights from ``cfg``.
projects/FairMOT/fairmot/build.py:17
↓ 1 callersFunctionbuild_transform_gen
(cfg, is_train)
projects/FairMOT/fairmot/dataset_mapper.py:16
↓ 1 callersMethodbuild_writers
Build a list of writers to be used. By default it contains writers that write metrics to the screen, a json file, and a tenso
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:166
↓ 1 callersMethodcompute_loss
(self, new_t1_feats_list, new_t2_feats_list, t1_info, t2_info, gt_ids_t2, edge_dicts)
projects/SGT/sgt/meta_arch/graphtracker.py:68
↓ 1 callersFunctioncompute_rot_loss
(output, target_bin, target_res, mask)
projects/CenterNet/centernet/losses.py:214
↓ 1 callersMethodconvert_all
(self)
projects/Datasets/CrowdHuman/gen_labels_crowdhuman.py:18
↓ 1 callersMethodconvert_all
(self)
projects/Datasets/Hieve/gen_labels_hieve.py:17
↓ 1 callersMethodconvert_all
(self)
projects/Datasets/Hieve/hieve2mot.py:37
↓ 1 callersMethodconvert_epoch2iter
(self, cfg)
projects/EpochTrainer/epoch_trainer/default_epoch_trainer.py:84
↓ 1 callersMethodconvert_weight_name
(self, checkpoint_dict, model_state_dict)
projects/SGT/sgt/checkpointer.py:158
↓ 1 callersMethodconvert_weight_name
(self, checkpoint_dict, model_state_dict)
projects/CenterNet/centernet/checkpoint/centernet_checkpoint.py:95
↓ 1 callersMethodcut_det_grad
(self, detector_outputs)
projects/SGT/sgt/meta_arch/graphtracker.py:85
↓ 1 callersMethoddecode
r""" decode output feature map to detection results Args: fmap(Tensor): output feature map hm(Tensor): output
projects/CenterNet/centernet/generator/centernet_decode.py:58
↓ 1 callersMethoddecode_gt
(results, batched_inputs)
projects/CenterNet/centernet/generator/centernet_decode.py:29
↓ 1 callersMethoddecode_gt_inference
(self, det_outs, batched_inputs)
projects/SGT/sgt/meta_arch/graphtracker_centernet.py:59
↓ 1 callersMethoddecode_gt_training
(self, targets, det_outs)
projects/SGT/sgt/meta_arch/graphtracker_centernet.py:98
↓ 1 callersMethoddecode_inference
(self, batched_inputs, detector_outs)
projects/SGT/sgt/meta_arch/graphtracker_centernet.py:43
↓ 1 callersMethoddecode_thresh
(self, results, tlbr_flag=True)
projects/SGT/sgt/meta_arch/graphtracker_centernet.py:216
↓ 1 callersMethoddecode_topk
(self, results, tlbr_flag=True)
projects/SGT/sgt/meta_arch/graphtracker_centernet.py:154
↓ 1 callersMethoddecode_training
(self, det_outs, detector_outs)
projects/SGT/sgt/meta_arch/graphtracker_centernet.py:34
↓ 1 callersMethoddraw_box
Args: box_coord (tuple): a tuple containing x0, y0, x1, y1 coordinates, where x0 and y0 are the coordinates of th
projects/Datasets/MOT/vis/simple_visualizer.py:352
↓ 1 callersMethoddraw_dataset_dict_simple
(self, dic)
projects/Datasets/MOT/vis/simple_visualizer.py:121
↓ 1 callersMethoddraw_gaussian
(fmap, center, radius, k=1)
projects/CenterNet/centernet/generator/centernet_gt.py:149
↓ 1 callersMethodeliminate_duplicated_prediction
(self, predictions)
projects/Datasets/MOT/evaluation/mot_evaluation.py:284
↓ 1 callersMethodeval_file
(self, seq_name)
projects/Datasets/MOT/evaluation/mot_evaluation.py:159
↓ 1 callersMethodeval_frame
(self, frame_id, trk_tlwhs, trk_ids, rtn_events=False)
projects/Datasets/MOT/evaluation/mot_evaluation.py:120
↓ 1 callersMethodevaluate_detections_by_coco_api_by_sequence
(self, predictions)
projects/Datasets/MOT/evaluation/mot_evaluation.py:338
↓ 1 callersMethodevaluation_tracking
(self)
projects/Datasets/MOT/evaluation/mot_evaluation.py:448
↓ 1 callersMethodextract_frames_one_video
(video_file_path, save_dir, img_dir)
projects/Datasets/Hieve/hieve2mot.py:115
↓ 1 callersFunctionf1
(covariance)
projects/FairMOT/fairmot/tracker/kalman_filter.py:276
↓ 1 callersFunctionf2
(covariance)
projects/FairMOT/fairmot/tracker/kalman_filter.py:279
↓ 1 callersFunctionf3
(covariance)
projects/FairMOT/fairmot/tracker/kalman_filter.py:282
← previousnext →101–200 of 586, ranked by callers