SYMBOL INDEX (138 symbols across 25 files) FILE: demo/demo.py function setup_cfg (line 20) | def setup_cfg(args): function get_parser (line 34) | def get_parser(): FILE: demo/predictor.py class VisualizationDemo (line 17) | class VisualizationDemo(object): method __init__ (line 18) | def __init__(self, cfg, instance_mode=ColorMode.IMAGE, parallel=False): method run_on_image (line 39) | def run_on_image(self, image): method _frame_from_video (line 73) | def _frame_from_video(self, video): method vis_bases (line 81) | def vis_bases(self, bases): method run_on_video (line 98) | def run_on_video(self, video): class AsyncPredictor (line 154) | class AsyncPredictor: class _StopToken (line 161) | class _StopToken: class _PredictWorker (line 164) | class _PredictWorker(mp.Process): method __init__ (line 165) | def __init__(self, cfg, task_queue, result_queue): method run (line 171) | def run(self): method __init__ (line 182) | def __init__(self, cfg, num_gpus: int = 1): method put (line 209) | def put(self, image): method get (line 213) | def get(self): method __len__ (line 229) | def __len__(self): method __call__ (line 232) | def __call__(self, image): method shutdown (line 236) | def shutdown(self): method default_buffer_size (line 241) | def default_buffer_size(self): FILE: fcos/checkpoint/adet_checkpoint.py class AdetCheckpointer (line 6) | class AdetCheckpointer(DetectionCheckpointer): method _load_file (line 11) | def _load_file(self, filename): FILE: fcos/config/config.py function get_cfg (line 4) | def get_cfg() -> CfgNode: FILE: fcos/data/builtin.py function register_all_coco (line 17) | def register_all_coco(root="datasets"): FILE: fcos/layers/conv_with_kaiming_uniform.py function conv_with_kaiming_uniform (line 8) | def conv_with_kaiming_uniform( FILE: fcos/layers/csrc/ml_nms/ml_nms.h function namespace (line 4) | namespace adet { FILE: fcos/layers/csrc/vision.cpp type adet (line 5) | namespace adet { function get_cuda_version (line 11) | std::string get_cuda_version() { function get_compiler_version (line 32) | std::string get_compiler_version() { function PYBIND11_MODULE (line 53) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: fcos/layers/deform_conv.py class _NewEmptyTensorOp (line 7) | class _NewEmptyTensorOp(torch.autograd.Function): method forward (line 9) | def forward(ctx, x, new_shape): method backward (line 14) | def backward(ctx, grad): class DFConv2d (line 19) | class DFConv2d(nn.Module): method __init__ (line 29) | def __init__( method forward (line 95) | def forward(self, x, return_offset=False): FILE: fcos/layers/iou_loss.py class IOULoss (line 5) | class IOULoss(nn.Module): method __init__ (line 14) | def __init__(self, loc_loss_type='iou'): method forward (line 18) | def forward(self, pred, target, weight=None): FILE: fcos/layers/ml_nms.py function ml_nms (line 4) | def ml_nms(boxlist, nms_thresh, max_proposals=-1, FILE: fcos/modeling/backbone/fpn.py class LastLevelP6P7 (line 12) | class LastLevelP6P7(nn.Module): method __init__ (line 18) | def __init__(self, in_channels, out_channels, in_features="res5"): method forward (line 27) | def forward(self, x): class LastLevelP6 (line 33) | class LastLevelP6(nn.Module): method __init__ (line 38) | def __init__(self, in_channels, out_channels, in_features="res5"): method forward (line 46) | def forward(self, x): function build_fcos_resnet_fpn_backbone (line 52) | def build_fcos_resnet_fpn_backbone(cfg, input_shape: ShapeSpec): FILE: fcos/modeling/backbone/mobilenet.py function conv_bn (line 13) | def conv_bn(inp, oup, stride): function conv_1x1_bn (line 21) | def conv_1x1_bn(inp, oup): class InvertedResidual (line 29) | class InvertedResidual(nn.Module): method __init__ (line 30) | def __init__(self, inp, oup, stride, expand_ratio): method forward (line 63) | def forward(self, x): class MobileNetV2 (line 70) | class MobileNetV2(Backbone): method __init__ (line 74) | def __init__(self, cfg, n_class=1000, input_size=224, width_mult=1.): method _freeze_backbone (line 110) | def _freeze_backbone(self, freeze_at): method forward (line 115) | def forward(self, x): method _initialize_weights (line 123) | def _initialize_weights(self): function build_mnv2_backbone (line 139) | def build_mnv2_backbone(cfg, input_shape): FILE: fcos/modeling/backbone/vovnet.py function conv3x3 (line 67) | def conv3x3(in_channels, out_channels, module_name, postfix, function conv1x1 (line 84) | def conv1x1(in_channels, out_channels, module_name, postfix, class Hsigmoid (line 100) | class Hsigmoid(nn.Module): method __init__ (line 101) | def __init__(self, inplace=True): method forward (line 105) | def forward(self, x): class eSEModule (line 109) | class eSEModule(nn.Module): method __init__ (line 110) | def __init__(self, channel, reduction=4): method forward (line 117) | def forward(self, x): class _OSA_module (line 125) | class _OSA_module(nn.Module): method __init__ (line 127) | def __init__(self, method forward (line 151) | def forward(self, x): class _OSA_stage (line 172) | class _OSA_stage(nn.Sequential): method __init__ (line 174) | def __init__(self, class VoVNet (line 211) | class VoVNet(Backbone): method __init__ (line 213) | def __init__(self, cfg, input_ch, out_features=None): method _initialize_weights (line 271) | def _initialize_weights(self): method _freeze_backbone (line 276) | def _freeze_backbone(self, freeze_at): method forward (line 292) | def forward(self, x): method output_shape (line 304) | def output_shape(self): function build_vovnet_backbone (line 314) | def build_vovnet_backbone(cfg, input_shape): function build_vovnet_fpn_backbone (line 326) | def build_vovnet_fpn_backbone(cfg, input_shape: ShapeSpec): function build_fcos_vovnet_fpn_backbone (line 349) | def build_fcos_vovnet_fpn_backbone(cfg, input_shape: ShapeSpec): FILE: fcos/modeling/fcos/fcos.py class Scale (line 19) | class Scale(nn.Module): method __init__ (line 20) | def __init__(self, init_value=1.0): method forward (line 24) | def forward(self, input): class FCOS (line 29) | class FCOS(nn.Module): method __init__ (line 33) | def __init__(self, cfg, input_shape: Dict[str, ShapeSpec]): method forward (line 63) | def forward(self, images, features, gt_instances): method compute_locations (line 120) | def compute_locations(self, features): method compute_locations_per_level (line 131) | def compute_locations_per_level(self, h, w, stride, device): class FCOSHead (line 147) | class FCOSHead(nn.Module): method __init__ (line 148) | def __init__(self, cfg, input_shape: List[ShapeSpec]): method forward (line 233) | def forward(self, x): FILE: fcos/modeling/fcos/fcos_outputs.py function aligned_bilinear (line 40) | def aligned_bilinear(tensor, factor): function compute_ctrness_targets (line 65) | def compute_ctrness_targets(reg_targets): class FCOSOutputs (line 75) | class FCOSOutputs(object): method __init__ (line 76) | def __init__( method _transpose (line 125) | def _transpose(self, training_targets, num_loc_list): method _get_ground_truth (line 142) | def _get_ground_truth(self): method get_sample_region (line 173) | def get_sample_region(self, gt, strides, num_loc_list, loc_xs, loc_ys,... method compute_targets_for_locations (line 205) | def compute_targets_for_locations(self, locations, targets, size_ranges): method losses (line 264) | def losses(self): method predict_proposals (line 337) | def predict_proposals(self): method forward_for_mask (line 366) | def forward_for_mask(self, boxlists): method forward_for_single_feature_map (line 396) | def forward_for_single_feature_map( method select_over_all_levels (line 466) | def select_over_all_levels(self, boxlists): method prepare_masks (line 487) | def prepare_masks(self, m_h, m_w, r_h, r_w, targets_masks): method dice_loss (line 505) | def dice_loss(self,input, target): method fcos_losses (line 514) | def fcos_losses( FILE: fcos/modeling/one_stage_detector.py class OneStageDetector (line 6) | class OneStageDetector(ProposalNetwork): method forward (line 11) | def forward(self, batched_inputs): FILE: fcos/modeling/poolers.py function _box_max_size (line 12) | def _box_max_size(boxes): function assign_boxes_to_levels_by_length (line 18) | def assign_boxes_to_levels_by_length( class TopPooler (line 51) | class TopPooler(ROIPooler): method __init__ (line 55) | def __init__(self, method forward (line 68) | def forward(self, x, box_lists): FILE: fcos/utils/comm.py function reduce_sum (line 5) | def reduce_sum(tensor): FILE: fcos/utils/measures.py function get_num_gen (line 13) | def get_num_gen(gen): function is_pruned (line 17) | def is_pruned(layer): function is_leaf (line 25) | def is_leaf(model): function get_layer_info (line 29) | def get_layer_info(layer): function get_layer_param (line 35) | def get_layer_param(model): function measure_layer (line 40) | def measure_layer(layer, *args): function measure_model (line 158) | def measure_model(model, x): FILE: postprocessing.py function detector_postprocess (line 8) | def detector_postprocess(results, output_height, output_width, mask_thre... function sem_seg_postprocess (line 57) | def sem_seg_postprocess(result, img_size, output_height, output_width): FILE: tools/compute_flops.py function setup (line 10) | def setup(args): function main (line 22) | def main(args): FILE: tools/convert_fcos_weight.py function get_parser (line 7) | def get_parser(): function rename_resnet_param_names (line 24) | def rename_resnet_param_names(ckpt_state_dict): FILE: tools/remove_optim_from_ckpt.py function get_parser (line 6) | def get_parser(): FILE: train_net.py class Trainer (line 45) | class Trainer(DefaultTrainer): method __init__ (line 51) | def __init__(self, cfg): method train_loop (line 86) | def train_loop(self, start_iter: int, max_iter: int): method train (line 105) | def train(self): method build_train_loader (line 118) | def build_train_loader(cls, cfg): method build_evaluator (line 130) | def build_evaluator(cls, cfg, dataset_name, output_folder=None): method test_with_TTA (line 175) | def test_with_TTA(cls, cfg, model): function setup (line 192) | def setup(args): function main (line 204) | def main(args):