SYMBOL INDEX (239 symbols across 32 files)

FILE: dataset/cocodataset.py
  class COCODataset (line 27) | class COCODataset(Dataset):
    method __init__ (line 31) | def __init__(self, data_dir='data/COCO', json_file='instances_train201...
    method __len__ (line 63) | def __len__(self):
    method pull_item (line 66) | def pull_item(self, index):
    method __getitem__ (line 121) | def __getitem__(self, index):

FILE: dataset/data_augment.py
  function _crop (line 19) | def _crop(image, boxes, labels, ratios = None):
  function _distort (line 86) | def _distort(image):
  function _expand (line 116) | def _expand(image, boxes,fill, p):
  function _mirror (line 152) | def _mirror(image, boxes):
  function _random_affine (line 161) | def _random_affine(img, targets=None, degrees=(-10, 10), translate=(.1, ...
  function preproc_for_test (line 228) | def preproc_for_test(image, input_size, mean, std):
  class TrainTransform (line 242) | class TrainTransform(object):
    method __init__ (line 244) | def __init__(self, p=0.5, rgb_means=None, std = None,max_labels=50):
    method __call__ (line 250) | def __call__(self, image, targets, input_dim):
  class ValTransform (line 361) | class ValTransform(object):
    method __init__ (line 376) | def __init__(self, rgb_means=None, std=None, swap=(2, 0, 1)):
    method __call__ (line 382) | def __call__(self, img, res, input_size):

FILE: dataset/dataloading.py
  class Dataset (line 14) | class Dataset(torchDataset):
    method __init__ (line 21) | def __init__(self, input_dimension):
    method input_dim (line 26) | def input_dim(self):
    method resize_getitem (line 38) | def resize_getitem(getitem_fn):
  class DataLoader (line 75) | class DataLoader(torchDataLoader):
    method __init__ (line 114) | def __init__(self, *args, **kwargs):
    method change_input_dim (line 157) | def change_input_dim(self, multiple=32, random_range=(10, 19)):
  class YoloBatchSampler (line 193) | class YoloBatchSampler(torchBatchSampler):
    method __init__ (line 198) | def __init__(self, *args, input_dimension=None, **kwargs):
    method __iter__ (line 203) | def __iter__(self):
    method __set_input_dim (line 209) | def __set_input_dim(self):
  class IterationBasedBatchSampler (line 216) | class IterationBasedBatchSampler(torchBatchSampler):
    method __init__ (line 222) | def __init__(self, batch_sampler, num_iterations, start_iter=0):
    method __iter__ (line 227) | def __iter__(self):
    method __len__ (line 241) | def __len__(self):
  function list_collate (line 244) | def list_collate(batch):

FILE: dataset/mixupdetection.py
  class MixupDetection (line 9) | class MixupDetection(Dataset):
    method __init__ (line 22) | def __init__(self, dataset, mixup=None, preproc=None, *args):
    method set_mixup (line 29) | def set_mixup(self, mixup=None, *args):
    method __len__ (line 42) | def __len__(self):
    method __getitem__ (line 46) | def __getitem__(self, idx):
    method pull_item (line 87) | def pull_item(self, idx):

FILE: dataset/voc_eval.py
  function parse_rec (line 14) | def parse_rec(filename):
  function voc_ap (line 35) | def voc_ap(rec, prec, use_07_metric=False):
  function voc_eval (line 68) | def voc_eval(detpath,

FILE: dataset/vocdataset.py
  class AnnotationTransform (line 40) | class AnnotationTransform(object):
    method __init__ (line 54) | def __init__(self, class_to_ind=None, keep_difficult=True):
    method __call__ (line 59) | def __call__(self, target):
  class VOCDetection (line 90) | class VOCDetection(Dataset):
    method __init__ (line 108) | def __init__(self, root, image_sets, preproc=None, target_transform=An...
    method __getitem__ (line 127) | def __getitem__(self, index):
    method __len__ (line 147) | def __len__(self):
    method pull_image (line 150) | def pull_image(self, index):
    method pull_anno (line 164) | def pull_anno(self, index):
    method pull_item (line 181) | def pull_item(self, index):
    method evaluate_detections (line 204) | def evaluate_detections(self, all_boxes, output_dir=None):
    method _get_voc_results_file_template (line 226) | def _get_voc_results_file_template(self):
    method _write_voc_results_file (line 235) | def _write_voc_results_file(self, all_boxes):
    method _do_python_eval (line 254) | def _do_python_eval(self, output_dir='output', iou = 0.5):

FILE: demo.py
  function parse_args (line 24) | def parse_args():
  function demo (line 42) | def demo():

FILE: eval.py
  function parse_args (line 34) | def parse_args():
  function eval (line 66) | def eval():

FILE: main.py
  function parse_args (line 37) | def parse_args():
  function main (line 86) | def main():

FILE: models/network_blocks.py
  function add_conv (line 7) | def add_conv(in_ch, out_ch, ksize, stride, leaky=True):
  class upsample (line 31) | class upsample(nn.Module):
    method __init__ (line 34) | def __init__(self, size=None, scale_factor=None, mode='nearest', align...
    method forward (line 42) | def forward(self, input):
    method extra_repr (line 45) | def extra_repr(self):
  class SPPLayer (line 53) | class SPPLayer(nn.Module):
    method __init__ (line 54) | def __init__(self):
    method forward (line 57) | def forward(self, x):
  class DropBlock (line 65) | class DropBlock(nn.Module):
    method __init__ (line 66) | def __init__(self, block_size=7, keep_prob=0.9):
    method reset (line 75) | def reset(self, block_size, keep_prob):
    method calculate_gamma (line 83) | def calculate_gamma(self, x):
    method forward (line 87) | def forward(self, x):
  class resblock (line 109) | class resblock(nn.Module):
    method __init__ (line 118) | def __init__(self, ch, nblocks=1, shortcut=True):
    method forward (line 129) | def forward(self, x):
  class RFBblock (line 138) | class RFBblock(nn.Module):
    method __init__ (line 139) | def __init__(self,in_ch,residual=False):
    method forward (line 161) | def forward(self,x):
  class FeatureAdaption (line 172) | class FeatureAdaption(nn.Module):
    method __init__ (line 173) | def __init__(self, in_ch, out_ch, n_anchors, rfb=False, sep=False):
    method forward (line 187) | def forward(self, input, wh_pred):
  class ASFFmobile (line 199) | class ASFFmobile(nn.Module):
    method __init__ (line 200) | def __init__(self, level, rfb=False, vis=False):
    method forward (line 228) | def forward(self, x_level_0, x_level_1, x_level_2):
  class ASFF (line 267) | class ASFF(nn.Module):
    method __init__ (line 268) | def __init__(self, level, rfb=False, vis=False):
    method forward (line 295) | def forward(self, x_level_0, x_level_1, x_level_2):
  function make_divisible (line 332) | def make_divisible(v, divisor, min_value=None):
  class ConvBNReLU (line 352) | class ConvBNReLU(nn.Sequential):
    method __init__ (line 353) | def __init__(self, in_planes, out_planes, kernel_size=3, stride=1, gro...
  function add_sepconv (line 361) | def add_sepconv(in_ch, out_ch, ksize, stride):
  class InvertedResidual (line 375) | class InvertedResidual(nn.Module):
    method __init__ (line 376) | def __init__(self, inp, oup, stride, expand_ratio):
    method forward (line 397) | def forward(self, x):
  class ressepblock (line 403) | class ressepblock(nn.Module):
    method __init__ (line 404) | def __init__(self, ch, out_ch, in_ch=None, shortcut=True):
    method forward (line 415) | def forward(self, x):

FILE: models/utils_loss.py
  class IOUWH_loss (line 7) | class IOUWH_loss(nn.Module): #used for anchor guiding
    method __init__ (line 8) | def __init__(self, reduction='none'):
    method forward (line 12) | def forward(self, pred, target):
  class IOUloss (line 39) | class IOUloss(nn.Module):
    method __init__ (line 40) | def __init__(self, reduction='none'):
    method forward (line 44) | def forward(self, pred, target):

FILE: models/yolov3_asff.py
  function build_yolov3_modules (line 9) | def build_yolov3_modules(num_classes, ignore_thre, label_smooth, rfb):
  class YOLOv3 (line 64) | class YOLOv3(nn.Module):
    method __init__ (line 70) | def __init__(self, num_classes = 80, ignore_thre=0.7, label_smooth = F...
    method forward (line 96) | def forward(self, x, targets=None, epoch=0):

FILE: models/yolov3_baseline.py
  function create_yolov3_modules (line 8) | def create_yolov3_modules(num_classes, ignore_thre, label_smooth, rfb):
  class YOLOv3 (line 74) | class YOLOv3(nn.Module):
    method __init__ (line 80) | def __init__(self, num_classes = 80, ignore_thre=0.7, label_smooth = F...
    method forward (line 85) | def forward(self, x, targets=None, epoch=0):

FILE: models/yolov3_head.py
  class YOLOv3Head (line 10) | class YOLOv3Head(nn.Module):
    method __init__ (line 11) | def __init__(self, anch_mask, n_classes, stride, in_ch=1024, ignore_th...
    method forward (line 49) | def forward(self, xin, labels=None):

FILE: models/yolov3_mobilev2.py
  function create_yolov3_mobilenet_v2 (line 6) | def create_yolov3_mobilenet_v2(num_classes, width_mult=1.0, inverted_res...
  class YOLOv3 (line 85) | class YOLOv3(nn.Module):
    method __init__ (line 91) | def __init__(self, num_classes = 80, ignore_thre=0.7, label_smooth = F...
    method forward (line 125) | def forward(self, x, targets=None, epoch=0):

FILE: utils/DCN/deform_conv2d_naive.py
  class deform_conv2d_naive (line 10) | class deform_conv2d_naive(Module):
    method __init__ (line 11) | def __init__(self, in_channels, out_channels,
    method reset_parameters (line 32) | def reset_parameters(self):
    method forward (line 40) | def forward(self, input, offset):
    method compute_mesh_grid (line 68) | def compute_mesh_grid(self, in_h, in_w):

FILE: utils/DCN/functions/deform_conv2d_func.py
  class DeformConv2dFunction (line 15) | class DeformConv2dFunction(Function):
    method forward (line 18) | def forward(ctx, input, offset, weight, bias,
    method backward (line 42) | def backward(ctx, grad_output):

FILE: utils/DCN/functions/modulated_deform_conv2d_func.py
  class ModulatedDeformConv2dFunction (line 15) | class ModulatedDeformConv2dFunction(Function):
    method forward (line 17) | def forward(ctx, input, offset, mask, weight, bias,
    method backward (line 40) | def backward(ctx, grad_output):

FILE: utils/DCN/modules/deform_conv2d.py
  class DeformConv2d (line 14) | class DeformConv2d(nn.Module):
    method __init__ (line 16) | def __init__(self, in_channels, out_channels,
    method reset_parameters (line 44) | def reset_parameters(self):
    method forward (line 52) | def forward(self, input, offset):
  class DeformConv2dPack (line 67) | class DeformConv2dPack(DeformConv2d):
    method __init__ (line 69) | def __init__(self, in_channels, out_channels,
    method init_offset (line 86) | def init_offset(self):
    method forward (line 90) | def forward(self, input):
  class DeformConv2dPackMore (line 103) | class DeformConv2dPackMore(DeformConv2d):
    method __init__ (line 105) | def __init__(self, in_channels, out_channels,
    method init_offset (line 122) | def init_offset(self):
    method forward (line 126) | def forward(self, input):

FILE: utils/DCN/modules/modulated_deform_conv2d.py
  class ModulatedDeformConv2d (line 14) | class ModulatedDeformConv2d(nn.Module):
    method __init__ (line 16) | def __init__(self, in_channels, out_channels,
    method reset_parameters (line 43) | def reset_parameters(self):
    method forward (line 51) | def forward(self, input, offset, mask):
  class ModulatedDeformConv2dPack (line 68) | class ModulatedDeformConv2dPack(ModulatedDeformConv2d):
    method __init__ (line 70) | def __init__(self, in_channels, out_channels,
    method init_offset (line 87) | def init_offset(self):
    method forward (line 91) | def forward(self, input):

FILE: utils/DCN/setup.py
  function get_extensions (line 17) | def get_extensions():

FILE: utils/DCN/src/cpu/deform_conv2d_cpu.cpp
  function deform_conv2d_cpu_forward (line 7) | at::Tensor
  function deform_conv2d_cpu_backward (line 27) | std::vector<at::Tensor>

FILE: utils/DCN/src/cpu/modulated_deform_conv2d_cpu.cpp
  function modulated_deform_conv2d_cpu_forward (line 7) | at::Tensor
  function modulated_deform_conv2d_cpu_backward (line 28) | std::vector<at::Tensor>

FILE: utils/DCN/src/vision.cpp
  function PYBIND11_MODULE (line 5) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: utils/cocoapi_evaluator.py
  function _accumulate_predictions_from_multiple_gpus (line 19) | def _accumulate_predictions_from_multiple_gpus(predictions_per_gpu):
  class COCOAPIEvaluator (line 31) | class COCOAPIEvaluator():
    method __init__ (line 37) | def __init__(self, data_dir, img_size, confthre, nmsthre, testset=Fals...
    method evaluate (line 74) | def evaluate(self, model, half=False, distributed=False):

FILE: utils/distributed_util.py
  function get_world_size (line 9) | def get_world_size():
  function get_rank (line 15) | def get_rank():
  function is_main_process (line 21) | def is_main_process():
  function synchronize (line 27) | def synchronize():
  function _encode (line 53) | def _encode(encoded_data, data):
  function _decode (line 67) | def _decode(encoded_data):
  function scatter_gather (line 75) | def scatter_gather(data):
  function reduce_loss_dict (line 140) | def reduce_loss_dict(loss_dict):

FILE: utils/fp16_utils/fp16_optimizer.py
  class FP16_Optimizer (line 11) | class FP16_Optimizer(object):
    method __init__ (line 105) | def __init__(self,
    method maybe_print (line 177) | def maybe_print(self, msg):
    method __getstate__ (line 181) | def __getstate__(self):
    method __setstate__ (line 184) | def __setstate__(self, state):
    method zero_grad (line 187) | def zero_grad(self, set_grads_to_None=False):
    method _check_overflow (line 213) | def _check_overflow(self):
    method _update_scale (line 223) | def _update_scale(self, has_overflow=False):
    method _master_params_to_model_params (line 226) | def _master_params_to_model_params(self):
    method _model_grads_to_master_grads (line 232) | def _model_grads_to_master_grads(self):
    method _downscale_master (line 236) | def _downscale_master(self):
    method clip_master_grads (line 243) | def clip_master_grads(self, max_norm, norm_type=2):
    method state_dict (line 267) | def state_dict(self):
    method load_state_dict (line 288) | def load_state_dict(self, state_dict):
    method step (line 330) | def step(self, closure=None): # could add clip option.
    method _step_with_closure (line 386) | def _step_with_closure(self, closure):
    method backward (line 425) | def backward(self, loss, update_master_grads=True, retain_graph=False):
    method update_master_grads (line 487) | def update_master_grads(self):
    method inspect_master_grad_data (line 500) | def inspect_master_grad_data(self):
    method _get_loss_scale (line 535) | def _get_loss_scale(self):
    method _set_loss_scale (line 538) | def _set_loss_scale(self, value):
    method _get_state (line 544) | def _get_state(self):
    method _set_state (line 547) | def _set_state(self, value):
    method _get_param_groups (line 554) | def _get_param_groups(self):
    method _set_param_groups (line 557) | def _set_param_groups(self, value):

FILE: utils/fp16_utils/fp16util.py
  class tofp16 (line 7) | class tofp16(nn.Module):
    method __init__ (line 15) | def __init__(self):
    method forward (line 18) | def forward(self, input):
  function BN_convert_float (line 22) | def BN_convert_float(module):
  function network_to_half (line 35) | def network_to_half(network):
  function convert_module (line 44) | def convert_module(module, dtype):
  function convert_network (line 60) | def convert_network(network, dtype):
  class FP16Model (line 71) | class FP16Model(nn.Module):
    method __init__ (line 76) | def __init__(self, network):
    method forward (line 80) | def forward(self, *inputs):
  function backwards_debug_hook (line 85) | def backwards_debug_hook(grad):
  function prep_param_lists (line 88) | def prep_param_lists(model, flat_master=False):
  function model_grads_to_master_grads (line 134) | def model_grads_to_master_grads(model_params, master_params, flat_master...
  function master_params_to_model_params (line 156) | def master_params_to_model_params(model_params, master_params, flat_mast...
  function to_python_float (line 174) | def to_python_float(t):

FILE: utils/fp16_utils/loss_scaler.py
  function to_python_float (line 4) | def to_python_float(t):
  class LossScaler (line 10) | class LossScaler:
    method __init__ (line 22) | def __init__(self, scale=1):
    method has_overflow (line 26) | def has_overflow(self, params):
    method _has_inf_or_nan (line 30) | def _has_inf_or_nan(x):
    method update_scale (line 33) | def update_scale(self, overflow):
    method loss_scale (line 37) | def loss_scale(self):
    method scale_gradient (line 40) | def scale_gradient(self, module, grad_in, grad_out):
    method backward (line 43) | def backward(self, loss, retain_graph=False):
  class DynamicLossScaler (line 47) | class DynamicLossScaler:
    method __init__ (line 73) | def __init__(self,
    method has_overflow (line 84) | def has_overflow(self, params):
    method _has_inf_or_nan (line 92) | def _has_inf_or_nan(x):
    method update_scale (line 113) | def update_scale(self, overflow):
    method loss_scale (line 124) | def loss_scale(self):
    method scale_gradient (line 127) | def scale_gradient(self, module, grad_in, grad_out):
    method backward (line 130) | def backward(self, loss, retain_graph=False):

FILE: utils/utils.py
  function postprocess (line 7) | def postprocess(prediction, num_classes, conf_thre=0.7, nms_thre=0.45):
  function bboxes_iou (line 72) | def bboxes_iou(bboxes_a, bboxes_b, xyxy=True):
  function matrix_iou (line 114) | def matrix_iou(a,b):
  function visual (line 126) | def visual(img, boxes, scores):

FILE: utils/vis_utils.py
  function make_vis (line 15) | def make_vis(dataset, index, img, fuse_weights, fused_fs):
  function make_pred_vis (line 34) | def make_pred_vis(dataset,index, img, class_names, bboxes, cls, scores):
  function vis (line 48) | def vis(img, boxes, scores, cls_ids, conf=0.5, class_names=None, color=N...
  function add_heat (line 91) | def add_heat(image, heat_map, max_v, min_v, alpha=0.4, save=None, cmap='...

FILE: utils/voc_evaluator.py
  function _accumulate_predictions_from_multiple_gpus (line 20) | def _accumulate_predictions_from_multiple_gpus(predictions_per_gpu):
  class VOCEvaluator (line 43) | class VOCEvaluator():
    method __init__ (line 49) | def __init__(self, data_dir, img_size, confthre, nmsthre,vis=False):
    method evaluate (line 75) | def evaluate(self, model, half=False):