SYMBOL INDEX (162 symbols across 22 files) FILE: bpm/dataset/Dataset.py class Dataset (line 6) | class Dataset(object): method __init__ (line 13) | def __init__( method set_mirror_type (line 38) | def set_mirror_type(self, mirror_type): method get_sample (line 41) | def get_sample(self, ptr): method next_batch (line 45) | def next_batch(self): method set_batch_size (line 49) | def set_batch_size(self, batch_size): method stop_prefetching_threads (line 55) | def stop_prefetching_threads(self): FILE: bpm/dataset/PreProcessImage.py class PreProcessIm (line 5) | class PreProcessIm(object): method __init__ (line 6) | def __init__( method __call__ (line 48) | def __call__(self, im): method check_mirror_type (line 52) | def check_mirror_type(mirror_type): method check_batch_dims (line 56) | def check_batch_dims(batch_dims): method set_mirror_type (line 61) | def set_mirror_type(self, mirror_type): method rand_crop_im (line 66) | def rand_crop_im(im, new_size, prng=np.random): method pre_process_im (line 76) | def pre_process_im(self, im): FILE: bpm/dataset/Prefetcher.py class Counter (line 6) | class Counter(object): method __init__ (line 9) | def __init__(self, val=0, max_val=0): method reset (line 14) | def reset(self): method set_max_value (line 18) | def set_max_value(self, max_val): method increment (line 21) | def increment(self): method get_value (line 30) | def get_value(self): class Enqueuer (line 35) | class Enqueuer(object): method __init__ (line 36) | def __init__(self, get_element, num_elements, num_threads=1, queue_siz... method start_ep (line 67) | def start_ep(self): method end_ep (line 71) | def end_ep(self): method reset (line 76) | def reset(self): method set_num_elements (line 88) | def set_num_elements(self, num_elements): method stop (line 93) | def stop(self): method enqueue (line 99) | def enqueue(self): class Prefetcher (line 122) | class Prefetcher(object): method __init__ (line 126) | def __init__(self, get_sample, dataset_size, batch_size, final_batch=T... method set_batch_size (line 151) | def set_batch_size(self, batch_size): method next_batch (line 160) | def next_batch(self): method start_ep_prefetching (line 190) | def start_ep_prefetching(self): method stop (line 197) | def stop(self): FILE: bpm/dataset/TestSet.py class TestSet (line 19) | class TestSet(Dataset): method __init__ (line 30) | def __init__( method set_feat_func (line 52) | def set_feat_func(self, extract_feat_func): method get_sample (line 55) | def get_sample(self, ptr): method next_batch (line 66) | def next_batch(self): method extract_feat (line 79) | def extract_feat(self, normalize_feat, verbose=True): method eval (line 131) | def eval( FILE: bpm/dataset/TrainSet.py class TrainSet (line 9) | class TrainSet(Dataset): method __init__ (line 14) | def __init__(self, method get_sample (line 25) | def get_sample(self, ptr): method next_batch (line 35) | def next_batch(self): FILE: bpm/dataset/__init__.py function create_dataset (line 12) | def create_dataset( FILE: bpm/model/PCBModel.py class PCBModel (line 9) | class PCBModel(nn.Module): method __init__ (line 10) | def __init__( method forward (line 42) | def forward(self, x): FILE: bpm/model/resnet.py function conv3x3 (line 17) | def conv3x3(in_planes, out_planes, stride=1, dilation=1): class BasicBlock (line 24) | class BasicBlock(nn.Module): method __init__ (line 27) | def __init__(self, inplanes, planes, stride=1, downsample=None, dilati... method forward (line 37) | def forward(self, x): class Bottleneck (line 56) | class Bottleneck(nn.Module): method __init__ (line 59) | def __init__(self, inplanes, planes, stride=1, downsample=None, dilati... method forward (line 72) | def forward(self, x): class ResNet (line 95) | class ResNet(nn.Module): method __init__ (line 97) | def __init__(self, block, layers, last_conv_stride=2, last_conv_dilati... method _make_layer (line 119) | def _make_layer(self, block, planes, blocks, stride=1, dilation=1): method forward (line 136) | def forward(self, x): function remove_fc (line 150) | def remove_fc(state_dict): function resnet18 (line 158) | def resnet18(pretrained=False, **kwargs): function resnet34 (line 170) | def resnet34(pretrained=False, **kwargs): function resnet50 (line 182) | def resnet50(pretrained=False, **kwargs): function resnet101 (line 194) | def resnet101(pretrained=False, **kwargs): function resnet152 (line 207) | def resnet152(pretrained=False, **kwargs): FILE: bpm/utils/dataset_utils.py function parse_im_name (line 10) | def parse_im_name(im_name, parse_type='id'): function get_im_names (line 20) | def get_im_names(im_dir, pattern='*.jpg', return_np=True, return_path=Fa... function move_ims (line 30) | def move_ims(ori_im_paths, new_im_dir, parse_im_name, new_im_name_tmpl): function partition_train_val_set (line 45) | def partition_train_val_set(im_names, parse_im_name, FILE: bpm/utils/distance.py function normalize (line 7) | def normalize(nparray, order=2, axis=0): function compute_dist (line 13) | def compute_dist(array1, array2, type='euclidean'): FILE: bpm/utils/metric.py function _unique_sample (line 15) | def _unique_sample(ids_dict, num): function cmc (line 23) | def cmc( function mean_ap (line 107) | def mean_ap( FILE: bpm/utils/re_ranking.py function re_ranking (line 35) | def re_ranking(q_g_dist, q_q_dist, g_g_dist, k1=20, k2=6, lambda_value=0... FILE: bpm/utils/utils.py function time_str (line 14) | def time_str(fmt=None): function load_pickle (line 20) | def load_pickle(path): function save_pickle (line 32) | def save_pickle(obj, path): function save_mat (line 39) | def save_mat(ndarray, path): function to_scalar (line 44) | def to_scalar(vt): function transfer_optim_state (line 55) | def transfer_optim_state(state, device_id=-1): function may_transfer_optims (line 80) | def may_transfer_optims(optims, device_id=-1): function may_transfer_modules_optims (line 93) | def may_transfer_modules_optims(modules_and_or_optims, device_id=-1): class TransferVarTensor (line 112) | class TransferVarTensor(object): method __init__ (line 115) | def __init__(self, device_id=-1): method __call__ (line 118) | def __call__(self, var_or_tensor): class TransferModulesOptims (line 123) | class TransferModulesOptims(object): method __init__ (line 126) | def __init__(self, device_id=-1): method __call__ (line 129) | def __call__(self, modules_and_or_optims): function set_devices (line 133) | def set_devices(sys_device_ids): function set_devices_for_ml (line 162) | def set_devices_for_ml(sys_device_ids): function load_ckpt (line 227) | def load_ckpt(modules_optims, ckpt_file, load_to_cpu=True, verbose=True): function save_ckpt (line 246) | def save_ckpt(modules_optims, ep, scores, ckpt_file): function load_state_dict (line 267) | def load_state_dict(model, src_state_dict): function is_iterable (line 308) | def is_iterable(obj): function may_set_mode (line 312) | def may_set_mode(maybe_modules, mode): function may_make_dir (line 325) | def may_make_dir(path): class AverageMeter (line 341) | class AverageMeter(object): method __init__ (line 345) | def __init__(self): method reset (line 351) | def reset(self): method update (line 357) | def update(self, val, n=1): class RunningAverageMeter (line 364) | class RunningAverageMeter(object): method __init__ (line 367) | def __init__(self, hist=0.99): method reset (line 372) | def reset(self): method update (line 376) | def update(self, val): class RecentAverageMeter (line 384) | class RecentAverageMeter(object): method __init__ (line 387) | def __init__(self, hist_size=100): method reset (line 392) | def reset(self): method update (line 396) | def update(self, val): method avg (line 403) | def avg(self): function get_model_wrapper (line 408) | def get_model_wrapper(model, multi_gpu): class ReDirectSTD (line 416) | class ReDirectSTD(object): method __init__ (line 436) | def __init__(self, fpath=None, console='stdout', immediately_visible=F... method __del__ (line 457) | def __del__(self): method __enter__ (line 460) | def __enter__(self): method __exit__ (line 463) | def __exit__(self, *args): method write (line 466) | def write(self, msg): method flush (line 478) | def flush(self): method close (line 485) | def close(self): function set_seed (line 491) | def set_seed(seed): function print_array (line 508) | def print_array(array, fmt='{:.2f}', end=' '): function str2bool (line 519) | def str2bool(v): function tight_float_str (line 523) | def tight_float_str(x, fmt='{:.4f}'): function find_index (line 527) | def find_index(seq, item): function adjust_lr_staircase (line 534) | def adjust_lr_staircase(param_groups, base_lrs, ep, decay_at_epochs, fac... function measure_time (line 574) | def measure_time(enter_msg, verbose=True): FILE: bpm/utils/visualization.py function add_border (line 9) | def add_border(im, border_width, value): function make_im_grid (line 31) | def make_im_grid(ims, n_rows, n_cols, space, pad_val): function get_rank_list (line 62) | def get_rank_list(dist_vec, q_id, q_cam, g_ids, g_cams, rank_list_size): function read_im (line 93) | def read_im(im_path): function save_im (line 105) | def save_im(im, save_path): function save_rank_list_to_im (line 112) | def save_rank_list_to_im(rank_list, same_id, q_im_path, g_im_paths, save... FILE: script/dataset/combine_trainval_sets.py function move_ims (line 23) | def move_ims( function combine_trainval_sets (line 49) | def combine_trainval_sets( FILE: script/dataset/mapping_im_names_duke.py function parse_original_im_name (line 18) | def parse_original_im_name(img_name, parse_type='id'): function map_im_names (line 28) | def map_im_names(ori_im_names, parse_im_name, new_im_name_tmpl): function save_im_name_mapping (line 42) | def save_im_name_mapping(raw_dir, ori_to_new_im_name_file): FILE: script/dataset/mapping_im_names_market1501.py function parse_original_im_name (line 18) | def parse_original_im_name(im_name, parse_type='id'): function map_im_names (line 29) | def map_im_names(ori_im_names, parse_im_name, new_im_name_tmpl): function save_im_name_mapping (line 43) | def save_im_name_mapping(raw_dir, ori_to_new_im_name_file): FILE: script/dataset/transform_cuhk03.py function save_images (line 26) | def save_images(mat_file, save_dir, new_im_name_tmpl): function transform (line 68) | def transform(zip_file, train_test_partition_file, save_dir=None): FILE: script/dataset/transform_duke.py function parse_original_im_name (line 24) | def parse_original_im_name(img_name, parse_type='id'): function save_images (line 34) | def save_images(zip_file, save_dir=None, train_test_split_file=None): function transform (line 75) | def transform(zip_file, save_dir=None): FILE: script/dataset/transform_market1501.py function parse_original_im_name (line 25) | def parse_original_im_name(im_name, parse_type='id'): function save_images (line 36) | def save_images(zip_file, save_dir=None, train_test_split_file=None): function transform (line 103) | def transform(zip_file, save_dir=None): FILE: script/experiment/train_pcb.py class Config (line 35) | class Config(object): method __init__ (line 36) | def __init__(self): class ExtractFeature (line 250) | class ExtractFeature(object): method __init__ (line 256) | def __init__(self, model, TVT): method __call__ (line 260) | def __call__(self, ims): function main (line 280) | def main(): FILE: script/experiment/visualize_rank_list.py class Config (line 31) | class Config(object): method __init__ (line 32) | def __init__(self): class ExtractFeature (line 140) | class ExtractFeature(object): method __init__ (line 146) | def __init__(self, model, TVT): method __call__ (line 150) | def __call__(self, ims): function main (line 170) | def main():