SYMBOL INDEX (98 symbols across 31 files) FILE: ret_benchmark/data/build.py function build_data (line 16) | def build_data(cfg, is_train=True): FILE: ret_benchmark/data/collate_batch.py function collate_fn (line 11) | def collate_fn(batch): FILE: ret_benchmark/data/datasets/base_dataset.py class BaseDataSet (line 16) | class BaseDataSet(Dataset): method __init__ (line 22) | def __init__(self, img_source, transforms=None, mode="RGB"): method __len__ (line 34) | def __len__(self): method __repr__ (line 37) | def __repr__(self): method __str__ (line 40) | def __str__(self): method _load_data (line 43) | def _load_data(self): method _build_label_index_dict (line 50) | def _build_label_index_dict(self): method __getitem__ (line 56) | def __getitem__(self, index): FILE: ret_benchmark/data/evaluations/ret_metric.py class RetMetric (line 11) | class RetMetric(object): method __init__ (line 12) | def __init__(self, feats, labels): method recall_k (line 31) | def recall_k(self, k=1): FILE: ret_benchmark/data/samplers/random_identity_sampler.py class RandomIdentitySampler (line 17) | class RandomIdentitySampler(Sampler): method __init__ (line 27) | def __init__(self, dataset, batch_size, num_instances, max_iters): method __len__ (line 35) | def __len__(self): method __repr__ (line 38) | def __repr__(self): method __str__ (line 41) | def __str__(self): method _prepare_batch (line 44) | def _prepare_batch(self): method __iter__ (line 58) | def __iter__(self): FILE: ret_benchmark/data/transforms/build.py function build_transforms (line 11) | def build_transforms(cfg, is_train=True): FILE: ret_benchmark/engine/trainer.py function do_train (line 20) | def do_train( FILE: ret_benchmark/losses/build.py function build_loss (line 13) | def build_loss(cfg): FILE: ret_benchmark/losses/margin_loss.py class DistanceWeightedSampling (line 9) | class DistanceWeightedSampling(object): method __init__ (line 12) | def __init__(self, cfg): method sample (line 17) | def sample(self, batch, labels): method p_dist (line 39) | def p_dist(A, eps=1e-4): method inverse_sphere_distances (line 45) | def inverse_sphere_distances(self, batch, dist, labels, anchor_label): class MarginLoss (line 65) | class MarginLoss(nn.Module): method __init__ (line 68) | def __init__(self, cfg): method forward (line 81) | def forward(self, batch, labels): FILE: ret_benchmark/losses/multi_similarity_loss.py class MultiSimilarityLoss (line 15) | class MultiSimilarityLoss(nn.Module): method __init__ (line 16) | def __init__(self, cfg): method forward (line 24) | def forward(self, feats, labels): FILE: ret_benchmark/modeling/backbone/bninception.py class BNInception (line 10) | class BNInception(nn.Module): method __init__ (line 12) | def __init__(self): method features (line 235) | def features(self, input): method logits (line 505) | def logits(self, features): method forward (line 510) | def forward(self, input): method load_param (line 515) | def load_param(self, model_path): FILE: ret_benchmark/modeling/backbone/build.py function build_backbone (line 7) | def build_backbone(cfg): FILE: ret_benchmark/modeling/backbone/resnet.py class ResNet50 (line 10) | class ResNet50(nn.Module): method __init__ (line 12) | def __init__(self): method forward (line 20) | def forward(self, x): method load_param (line 36) | def load_param(self, model_path): FILE: ret_benchmark/modeling/build.py function build_model (line 19) | def build_model(cfg): FILE: ret_benchmark/modeling/heads/build.py function build_head (line 13) | def build_head(cfg): FILE: ret_benchmark/modeling/heads/linear_norm.py class LinearNorm (line 16) | class LinearNorm(nn.Module): method __init__ (line 17) | def __init__(self, cfg, in_channels): method forward (line 22) | def forward(self, x): FILE: ret_benchmark/modeling/xbm.py class XBM (line 13) | class XBM: method __init__ (line 14) | def __init__(self, cfg, model): method enqueue_dequeue (line 35) | def enqueue_dequeue(self, feats, indices): method get (line 43) | def get(self): FILE: ret_benchmark/solver/build.py function build_optimizer (line 6) | def build_optimizer(cfg, model): function build_lr_scheduler (line 21) | def build_lr_scheduler(cfg, optimizer): FILE: ret_benchmark/solver/lr_scheduler.py class WarmupMultiStepLR (line 6) | class WarmupMultiStepLR(torch.optim.lr_scheduler._LRScheduler): method __init__ (line 7) | def __init__( method get_lr (line 35) | def get_lr(self): FILE: ret_benchmark/utils/checkpoint.py class Checkpointer (line 9) | class Checkpointer(object): method __init__ (line 10) | def __init__( method save (line 28) | def save(self, name): method load (line 43) | def load(self, f=None): method has_checkpoint (line 64) | def has_checkpoint(self): method get_checkpoint_file (line 68) | def get_checkpoint_file(self): method tag_last_checkpoint (line 80) | def tag_last_checkpoint(self, last_filename): method _load_file (line 85) | def _load_file(self, f): method _load_model (line 88) | def _load_model(self, checkpoint): FILE: ret_benchmark/utils/config_util.py function get_config_root_path (line 10) | def get_config_root_path(): function load_config (line 19) | def load_config(rel_path): function load_config_from_file (line 25) | def load_config_from_file(file_path): FILE: ret_benchmark/utils/feat_extractor.py function feat_extractor (line 12) | def feat_extractor(model, data_loader, logger=None): FILE: ret_benchmark/utils/freeze_bn.py function set_bn_eval (line 11) | def set_bn_eval(m): FILE: ret_benchmark/utils/img_reader.py function read_image (line 5) | def read_image(img_path, mode='RGB'): FILE: ret_benchmark/utils/init_methods.py function weights_init_kaiming (line 12) | def weights_init_kaiming(m): function weights_init_classifier (line 27) | def weights_init_classifier(m): FILE: ret_benchmark/utils/logger.py function setup_logger (line 17) | def setup_logger(name: str, level: int, stream: str = "stdout") -> loggi... FILE: ret_benchmark/utils/metric_logger.py class SmoothedValue (line 8) | class SmoothedValue(object): method __init__ (line 13) | def __init__(self, window_size=20): method update (line 19) | def update(self, value): method median (line 26) | def median(self): method avg (line 31) | def avg(self): method global_avg (line 36) | def global_avg(self): class MetricLogger (line 40) | class MetricLogger(object): method __init__ (line 41) | def __init__(self, delimiter="\t"): method update (line 45) | def update(self, **kwargs): method __getattr__ (line 52) | def __getattr__(self, attr): method __str__ (line 60) | def __str__(self): FILE: ret_benchmark/utils/model_serialization.py function align_and_update_state_dicts (line 8) | def align_and_update_state_dicts(model_state_dict, loaded_state_dict): function strip_prefix_if_present (line 59) | def strip_prefix_if_present(state_dict, prefix): function load_state_dict (line 69) | def load_state_dict(model, loaded_state_dict): FILE: ret_benchmark/utils/registry.py function _register_generic (line 4) | def _register_generic(module_dict, module_name, module): class Registry (line 9) | class Registry(dict): method __init__ (line 32) | def __init__(self, *args, **kwargs): method register (line 35) | def register(self, module_name, module=None): FILE: scripts/split_cub_for_ms_loss.py function main (line 8) | def main(): FILE: tools/main.py function train (line 21) | def train(cfg): function parse_args (line 61) | def parse_args():