SYMBOL INDEX (186 symbols across 23 files)

FILE: Evaluation/eval_detection.py
  function setup_logger (line 23) | def setup_logger(log_file_path, name=None, level=logging.INFO):
  function get_classes (line 59) | def get_classes(anno_dict):
  class ANETdetection (line 72) | class ANETdetection(object):
    method __init__ (line 77) | def __init__(self, ground_truth_filename=None, prediction_filename=None,
    method _import_ground_truth (line 119) | def _import_ground_truth(self, ground_truth_filename):
    method _import_prediction (line 174) | def _import_prediction(self, prediction_filename):
    method _get_predictions_with_label (line 231) | def _get_predictions_with_label(self, prediction_by_label, label_name,...
    method wrapper_compute_average_precision (line 242) | def wrapper_compute_average_precision(self):
    method evaluate (line 264) | def evaluate(self):
  function compute_average_precision_detection (line 276) | def compute_average_precision_detection(ground_truth, prediction, tiou_t...

FILE: Evaluation/utils.py
  function get_blocked_videos (line 8) | def get_blocked_videos(api=API):
  function interpolated_prec_rec (line 14) | def interpolated_prec_rec(prec, rec):
  function segment_iou (line 25) | def segment_iou(target_segment, candidate_segments):
  function wrapper_segment_iou (line 53) | def wrapper_segment_iou(target_segments, candidate_segments):

FILE: datasets/__init__.py
  function build_dataset (line 4) | def build_dataset(subset, args, mode):

FILE: datasets/data_utils.py
  function load_json (line 18) | def load_json(path):
  function get_valid_anno (line 22) | def get_valid_anno(gt_instances, slice, thr=0.75,
  function get_dataset_dict (line 45) | def get_dataset_dict(video_info_path, video_anno_path, subset, mode='tes...
  function load_video_frames (line 159) | def load_video_frames(frame_dir, start, seq_len, stride=1, fn_tmpl='img_...
  function load_feature (line 163) | def load_feature(ft_path, ft_format, shape=None):
  function get_dataset_info (line 176) | def get_dataset_info(dataset, feature):
  function make_img_transform (line 208) | def make_img_transform(*args, **kwargs):

FILE: datasets/tad_dataset.py
  class TADDataset (line 27) | class TADDataset(torch.utils.data.Dataset):
    method __init__ (line 28) | def __init__(self, subset, mode, feature_info, ann_file, ft_info_file,...
    method _get_classes (line 62) | def _get_classes(self, anno_dict):
    method _prepare (line 74) | def _prepare(self):
    method __len__ (line 102) | def __len__(self):
    method _get_video_data (line 105) | def _get_video_data(self, index):
    method _get_feature_data (line 111) | def _get_feature_data(self,index):
    method _get_img_data (line 157) | def _get_img_data(self, index):
    method _get_train_label (line 161) | def _get_train_label(self, video_name):
    method __getitem__ (line 202) | def __getitem__(self, index):
  function build (line 212) | def build(dataset, subset, args, mode):

FILE: datasets/tad_eval.py
  function eval_ap (line 27) | def eval_ap(iou, cls, gt, predition):
  function apply_nms (line 33) | def apply_nms(dets_arr, nms_thr=0.4, use_soft_nms=False):
  class TADEvaluator (line 52) | class TADEvaluator(object):
    method __init__ (line 53) | def __init__(self, dataset_name, subset, video_dict=None, nms_mode=['r...
    method _get_classes (line 108) | def _get_classes(self, anno_dict):
    method update (line 120) | def update(self, pred, assign_cls_labels=False):
    method nms_whole_dataset (line 167) | def nms_whole_dataset(self):
    method cross_window_fusion (line 178) | def cross_window_fusion(self):
    method accumulate (line 232) | def accumulate(self, test_slice_overlap=0):
    method import_prediction (line 247) | def import_prediction(self):
    method format_arr (line 250) | def format_arr(self, arr, format='{:.2f}'):
    method synchronize_between_processes (line 254) | def synchronize_between_processes(self):
    method summarize (line 262) | def summarize(self):
    method compute_map (line 297) | def compute_map(self, nms_mode):
    method dump_to_json (line 329) | def dump_to_json(self, dets, save_path):
    method dump_detection (line 356) | def dump_detection(self, save_path=None):
  function merge_distributed (line 363) | def merge_distributed(all_pred):

FILE: demo.py
  function demo (line 16) | def demo(args, cfg):

FILE: engine.py
  function train_one_epoch (line 24) | def train_one_epoch(model: torch.nn.Module, criterion: torch.nn.Module,
  function to_device (line 90) | def to_device(t, device):
  function test (line 98) | def test(model, criterion, postprocessor, data_loader, base_ds, device, ...

FILE: main.py
  function main (line 39) | def main(args):

FILE: models/__init__.py
  function build_model (line 10) | def build_model(args):

FILE: models/custom_loss.py
  function sigmoid_focal_loss (line 11) | def sigmoid_focal_loss(inputs, targets, num_boxes, alpha: float = 0.25, ...

FILE: models/matcher.py
  class HungarianMatcher (line 22) | class HungarianMatcher(nn.Module):
    method __init__ (line 29) | def __init__(self, cost_class: float = 1, cost_seg: float = 1, cost_io...
    method forward (line 43) | def forward(self, outputs, targets):
  function build_matcher (line 94) | def build_matcher(args):

FILE: models/ops/roi_align/roi_align.py
  class _Align1D (line 10) | class _Align1D(Function):
    method forward (line 12) | def forward(ctx, input, roi, feature_dim, ratio):
    method backward (line 24) | def backward(ctx, grad_output):
  class ROIAlign (line 44) | class ROIAlign(nn.Module):
    method __init__ (line 45) | def __init__(self, feature_dim, ratio=0):
    method forward (line 50) | def forward(self, input, rois):
    method __repr__ (line 64) | def __repr__(self):

FILE: models/ops/roi_align/src/roi_align_cuda.cpp
  function Align_forward (line 22) | at::Tensor Align_forward(const at::Tensor& input, // (bs,ch,t)
  function Align_backward (line 29) | at::Tensor Align_backward(const at::Tensor& grad,
  function PYBIND11_MODULE (line 39) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: models/ops/setup.py
  function get_sources (line 28) | def get_sources(extensions_dir):
  function get_extensions (line 35) | def get_extensions():

FILE: models/ops/temporal_deform_attn/temporal_deform_attn.py
  function _is_power_of_2 (line 32) | def _is_power_of_2(n):
  class DeformAttn (line 39) | class DeformAttn(nn.Module):
    method __init__ (line 40) | def __init__(self, d_model=256, n_levels=1, n_heads=8, n_points=4):
    method _reset_parameters (line 76) | def _reset_parameters(self):
    method forward (line 98) | def forward(self, query, reference_points, input_flatten, input_tempor...
  function deform_attn_core_pytorch (line 160) | def deform_attn_core_pytorch(value, value_spatial_shapes, sampling_locat...

FILE: models/position_encoding.py
  class PositionEmbeddingSine (line 20) | class PositionEmbeddingSine(nn.Module):
    method __init__ (line 25) | def __init__(self, num_pos_feats=256, temperature=10000, normalize=Fal...
    method forward (line 36) | def forward(self, tensor_list: NestedTensor):
  function build_position_encoding (line 56) | def build_position_encoding(args):

FILE: models/tadtr.py
  function _get_clones (line 37) | def _get_clones(module, N):
  function get_norm (line 41) | def get_norm(norm_type, dim, num_groups=None):
  class TadTR (line 51) | class TadTR(nn.Module):
    method __init__ (line 54) | def __init__(self, position_embedding, transformer, num_classes, num_q...
    method _to_roi_align_format (line 124) | def _to_roi_align_format(self, rois, T, scale_factor=1):
    method forward (line 145) | def forward(self, samples):
    method _set_aux_loss (line 228) | def _set_aux_loss(self, outputs_class, outputs_coord):
  class SetCriterion (line 236) | class SetCriterion(nn.Module):
    method __init__ (line 243) | def __init__(self, num_classes, matcher, weight_dict, losses, focal_al...
    method loss_labels (line 259) | def loss_labels(self, outputs, targets, indices, num_segments, log=True):
    method loss_segments (line 286) | def loss_segments(self, outputs, targets, indices, num_segments):
    method loss_actionness (line 307) | def loss_actionness(self, outputs, targets, indices, num_segments):
    method _get_src_permutation_idx (line 330) | def _get_src_permutation_idx(self, indices):
    method _get_tgt_permutation_idx (line 336) | def _get_tgt_permutation_idx(self, indices):
    method get_loss (line 342) | def get_loss(self, loss, outputs, targets, indices, num_segments, **kw...
    method forward (line 352) | def forward(self, outputs, targets):
  class PostProcess (line 398) | class PostProcess(nn.Module):
    method forward (line 402) | def forward(self, outputs, target_sizes, fuse_score=True):
  class MLP (line 453) | class MLP(nn.Module):
    method __init__ (line 456) | def __init__(self, input_dim, hidden_dim, output_dim, num_layers):
    method forward (line 463) | def forward(self, x):
  function build (line 469) | def build(args):

FILE: models/transformer.py
  class DeformableTransformer (line 25) | class DeformableTransformer(nn.Module):
    method __init__ (line 26) | def __init__(self, d_model=256, nhead=8,
    method _reset_parameters (line 51) | def _reset_parameters(self):
    method get_valid_ratio (line 63) | def get_valid_ratio(self, mask):
    method forward (line 69) | def forward(self, srcs, masks, pos_embeds, query_embed=None):
  class DeformableTransformerEncoderLayer (line 126) | class DeformableTransformerEncoderLayer(nn.Module):
    method __init__ (line 127) | def __init__(self,
    method with_pos_embed (line 147) | def with_pos_embed(tensor, pos):
    method forward_ffn (line 150) | def forward_ffn(self, src):
    method forward (line 156) | def forward(self, src, pos, reference_points, spatial_shapes, level_st...
  class DeformableTransformerEncoder (line 168) | class DeformableTransformerEncoder(nn.Module):
    method __init__ (line 169) | def __init__(self, encoder_layer, num_layers):
    method get_reference_points (line 175) | def get_reference_points(spatial_shapes, valid_ratios, device):
    method forward (line 185) | def forward(self, src, temporal_lens, level_start_index, valid_ratios,...
  class DeformableTransformerDecoderLayer (line 200) | class DeformableTransformerDecoderLayer(nn.Module):
    method __init__ (line 201) | def __init__(self, d_model=256, d_ffn=1024,
    method with_pos_embed (line 225) | def with_pos_embed(tensor, pos):
    method forward_ffn (line 228) | def forward_ffn(self, tgt):
    method forward (line 234) | def forward(self, tgt, query_pos, reference_points, src, src_spatial_s...
  class DeformableTransformerDecoder (line 258) | class DeformableTransformerDecoder(nn.Module):
    method __init__ (line 259) | def __init__(self, decoder_layer, num_layers, return_intermediate=False):
    method forward (line 268) | def forward(self, tgt, reference_points, src, src_spatial_shapes, src_...
  function _get_clones (line 310) | def _get_clones(module, N):
  function _get_activation_fn (line 314) | def _get_activation_fn(activation):
  function build_deformable_transformer (line 327) | def build_deformable_transformer(args):

FILE: opts.py
  function str2bool (line 13) | def str2bool(x):
  function get_args_parser (line 20) | def get_args_parser():
  function update_cfg_with_args (line 190) | def update_cfg_with_args(cfg, arg_list):
  function update_cfg_from_file (line 205) | def update_cfg_from_file(cfg, cfg_path):

FILE: util/logger.py
  function _suppress_print (line 13) | def _suppress_print():
  function setup_logger (line 24) | def setup_logger(log_file_path, name=None, level=logging.INFO):

FILE: util/misc.py
  function mkdir_if_not_exist (line 31) | def mkdir_if_not_exist(dirname):
  class SmoothedValue (line 36) | class SmoothedValue(object):
    method __init__ (line 41) | def __init__(self, window_size=20, fmt=None):
    method update (line 49) | def update(self, value, n=1):
    method synchronize_between_processes (line 54) | def synchronize_between_processes(self):
    method median (line 69) | def median(self):
    method avg (line 74) | def avg(self):
    method global_avg (line 79) | def global_avg(self):
    method max (line 83) | def max(self):
    method value (line 87) | def value(self):
    method __str__ (line 90) | def __str__(self):
  function all_gather (line 99) | def all_gather(data):
  function reduce_dict (line 144) | def reduce_dict(input_dict, average=True):
  class MetricLogger (line 171) | class MetricLogger(object):
    method __init__ (line 172) | def __init__(self, delimiter="\t"):
    method update (line 176) | def update(self, **kwargs):
    method __getattr__ (line 183) | def __getattr__(self, attr):
    method __str__ (line 191) | def __str__(self):
    method synchronize_between_processes (line 199) | def synchronize_between_processes(self):
    method add_meter (line 203) | def add_meter(self, name, meter):
    method log_every (line 206) | def log_every(self, iterable, print_freq, header=None):
  function get_sha (line 261) | def get_sha():
  function collate_fn (line 281) | def collate_fn(batch):
  function _max_by_axis (line 289) | def _max_by_axis(the_list):
  class NestedTensor (line 298) | class NestedTensor(object):
    method __init__ (line 299) | def __init__(self, tensors, mask: Optional[Tensor]):
    method to (line 303) | def to(self, device):
    method decompose (line 319) | def decompose(self):
    method __repr__ (line 322) | def __repr__(self):
  function nested_tensor_from_tensor_list (line 326) | def nested_tensor_from_tensor_list(tensor_list: List[Tensor]):
  function make_nested_tensor (line 370) | def make_nested_tensor(tensor):
  function _onnx_nested_tensor_from_tensor_list (line 379) | def _onnx_nested_tensor_from_tensor_list(tensor_list: List[Tensor]) -> N...
  function setup_for_distributed (line 410) | def setup_for_distributed(is_master):
  function is_dist_avail_and_initialized (line 425) | def is_dist_avail_and_initialized():
  function get_world_size (line 433) | def get_world_size():
  function get_rank (line 439) | def get_rank():
  function is_main_process (line 445) | def is_main_process():
  function save_on_master (line 449) | def save_on_master(*args, **kwargs):
  function init_distributed_mode (line 454) | def init_distributed_mode(args):
  function accuracy (line 480) | def accuracy(output, target, topk=(1,)):
  function inverse_sigmoid (line 498) | def inverse_sigmoid(x, eps=1e-5):

FILE: util/segment_ops.py
  function segment_cw_to_t1t2 (line 17) | def segment_cw_to_t1t2(x):
  function segment_t1t2_to_cw (line 34) | def segment_t1t2_to_cw(x):
  function segment_length (line 51) | def segment_length(segments):
  function segment_iou_and_union (line 56) | def segment_iou_and_union(segments1, segments2):
  function segment_iou (line 70) | def segment_iou(segments1, segments2):
  function temporal_iou_numpy (line 97) | def temporal_iou_numpy(proposal_min, proposal_max, gt_min, gt_max):
  function temporal_iou_numpy (line 118) | def temporal_iou_numpy(proposal_min, proposal_max, gt_min, gt_max):
  function soft_nms (line 141) | def soft_nms(proposals, alpha, low_threshold, high_threshold, top_k):
  function temporal_nms (line 190) | def temporal_nms(segments, thresh):