MCPcopy Create free account

hub / github.com/BADBADBADBOY/DBnet-lite.pytorch / functions

Functions187 in github.com/BADBADBADBOY/DBnet-lite.pytorch

Method__init__
(self)
dataloader/random_thansform.py:158
Method__init__
(self,shrink_ratio=0.4,thresh_min=0.3,thresh_max=0.7)
dataloader/MakeBorderMap.py:17
Method__init__
(self, config)
dataloader/dataload.py:47
Method__init__
(self, min_text_size = 8,shrink_ratio = 0.4,is_training = True)
dataloader/MakeSegMap.py:21
Method__init__
(self, config,is_train = True)
models/DBNet.py:15
Method__init__
(self, inplanes, planes, stride=1, downsample=None, dcn=None)
models/backbone/resnet.py:43
Method__init__
(self, inplanes, planes, stride=1, downsample=None, dcn=None)
models/backbone/resnet.py:115
Method__init__
(self, spatial_scale, out_size, out_channels,
models/dcn/modules/deform_pool.py:38
Method__init__
(self, spatial_scale, out_size, out_channels,
models/dcn/modules/deform_pool.py:97
Method__init__
(self, *args, **kwargs)
models/dcn/modules/deform_conv.py:61
Method__init__
(self, in_channels, out_channels, kernel_size,
models/dcn/modules/deform_conv.py:86
Method__init__
(self, *args, **kwargs)
models/dcn/modules/deform_conv.py:133
Method__init__
(self,in_channels,out_channels,kernel_size,stride,padding,bias)
models/head/DB_Head.py:14
Method__init__
(self,in_channels,out_channels,kernel_size,stride,padding)
models/head/FPN_Head.py:15
Method__init__
bias: Whether conv layers have bias or not. adaptive: Whether to use adaptive threshold training or not. smooth: If true, use
models/head/seg_detector.py:20
Method__init__
(self, eps=1e-6, l1_scale=10, bce_scale=1)
loss/loss.py:23
Method__init__
(self, eps=1e-6, coverage_scale=5.0)
loss/dice_loss.py:68
Method__init__
(self, threshold=0.3, iou_thresh=0.2, reduction=None, max_regions=100, eps=1e-6)
loss/dice_loss.py:102
Method__init__
(self, negative_ratio=3.)
loss/l1_loss.py:19
Method__init__
(self, negative_ratio=3.0, eps=1e-6)
loss/balance_cross_entropy_loss.py:24
Method__len__
(self)
dataloader/dataload.py:64
Methodbackward
(ctx, grad_output)
models/dcn/functions/deform_pool.py:50
Methodbackward
(ctx, grad_output)
models/dcn/functions/deform_conv.py:146
Functiondefault_evaluation_params
default_evaluation_params: Default parameters to use for the validation and evaluation.
cal_rescall/script.py:8
Functiondeform_conv_backward_input_cuda
models/dcn/src/deform_conv_cuda.cpp:260
Functiondeform_conv_backward_parameters_cuda
models/dcn/src/deform_conv_cuda.cpp:373
Functiondeform_conv_forward_cuda
models/dcn/src/deform_conv_cuda.cpp:151
Functiondeform_psroi_pooling_cuda_backward
models/dcn/src/deform_pool_cuda.cpp:54
Functiondeform_psroi_pooling_cuda_forward
models/dcn/src/deform_pool_cuda.cpp:30
Functiondeformable_resnet18
Constructs a ResNet-18 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/backbone/resnet.py:268
Functiondeformable_resnet50
Constructs a ResNet-50 model with deformable conv. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/backbone/resnet.py:308
Functiondice_loss
(input, target, mask,eps=1e-6)
loss/dice_loss.py:46
Functionevaluate_method
Method evaluate_method: evaluate method and returns the results Results. Dictionary with the following values: - method (required
cal_rescall/script.py:47
Methodextend_line
(self, point_1, point_2, result)
dataloader/MakeBorderMap.py:108
Methodforward
(self, x)
utils/fuse_model.py:23
Methodforward
(self, x)
models/DBNet.py:25
Methodforward
(self, x)
models/backbone/resnet.py:83
Methodforward
(self, x)
models/backbone/resnet.py:154
Methodforward
(self, x)
models/backbone/resnet.py:242
Methodforward
(self, data, rois, offset)
models/dcn/modules/deform_pool.py:27
Methodforward
(self, data, rois)
models/dcn/modules/deform_pool.py:72
Methodforward
(self, data, rois)
models/dcn/modules/deform_pool.py:150
Methodforward
(self, x, offset)
models/dcn/modules/deform_conv.py:54
Methodforward
(self, x)
models/dcn/modules/deform_conv.py:78
Methodforward
(self, x, offset, mask)
models/dcn/modules/deform_conv.py:125
Methodforward
(self, x)
models/dcn/modules/deform_conv.py:150
Methodforward
(ctx, data, rois, offset, spatial_scale,
models/dcn/functions/deform_pool.py:10
Methodforward
(ctx, input, offset, weight, stride=1,
models/dcn/functions/deform_conv.py:11
Methodforward
(ctx, input, offset, mask, weight,
models/dcn/functions/deform_conv.py:111
Methodforward
(self, x)
models/head/DB_Head.py:19
Methodforward
(self, x)
models/head/DB_Head.py:56
Methodforward
(self, x)
models/head/FPN_Head.py:20
Methodforward
(self, x)
models/head/FPN_Head.py:60
Methodforward
(self, features, gt=None, masks=None, training=False)
models/head/seg_detector.py:99
Methodforward
(self, pred, batch)
loss/loss.py:32
Methodforward
pred: one or two heatmaps of shape (N, 1, H, W), the losses of tow heatmaps are added together. gt: (N, 1, H, W)
loss/dice_loss.py:19
Methodforward
(self, pred, gt, mask)
loss/dice_loss.py:73
Methodforward
(self, pred, gt, mask)
loss/dice_loss.py:149
Methodforward
(self, pred: torch.Tensor, gt, mask)
loss/l1_loss.py:9
Methodforward
Args: pred: (N, 1, H, W). gt: (N, H, W). mask: (N, H, W).
loss/l1_loss.py:23
Methodforward
Args: pred: shape :math:`(N, 1, H, W)`, the prediction of network gt: shape :math:`(N, 1, H, W)`, the target
loss/balance_cross_entropy_loss.py:29
Functionget_tl_line_values_from_file_contents
Returns all points, confindences and transcriptions of a file in lists. Valid line formats: xmin,ymin,xmax,ymax,[confidence],[transcription]
cal_rescall/rrc_evaluation_funcs.py:281
Methodis_poly_in_rect
(self, poly, x, y, w, h)
dataloader/random_thansform.py:66
Functionload_folder_file
Returns an array with the contents (filtered by fileNameRegExp) of a ZIP file. The key's are the names or the file or the capturing group def
cal_rescall/rrc_evaluation_funcs.py:79
Functionload_zip_file
Returns an array with the contents (filtered by fileNameRegExp) of a ZIP file. The key's are the names or the file or the capturing group def
cal_rescall/rrc_evaluation_funcs.py:47
Functionload_zip_file_keys
Returns an array with the entries of the ZIP file that match with the regular expression. The key's are the names or the file or the capturin
cal_rescall/rrc_evaluation_funcs.py:18
Functionmain_evaluation
This process validates a method, evaluates it and if it succed generates a ZIP file with a JSON entry for each sample. Params: p: Diction
cal_rescall/rrc_evaluation_funcs.py:309
Functionmain_validation
This process validates a method Params: default_evaluation_params_fn: points to a function that returns a dictionary with the default par
cal_rescall/rrc_evaluation_funcs.py:371
Functionmodulated_deform_conv_cuda_backward
models/dcn/src/deform_conv_cuda.cpp:566
Functionmodulated_deform_conv_cuda_forward
models/dcn/src/deform_conv_cuda.cpp:486
Functionprint_help
()
cal_rescall/rrc_evaluation_funcs.py:13
Methodprocess
requied keys: image, polygons, ignore_tags, filename adding keys: mask
dataloader/MakeSegMap.py:25
Methodrandom_crop_pse
(self,imgs, img_size=(640, 640))
dataloader/random_thansform.py:218
Functionrectangle_to_points
(rect)
cal_rescall/script.py:86
Methodreset
(self)
utils/metrics.py:49
Functionresnet101
Constructs a ResNet-101 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/backbone/resnet.py:325
Functionresnet152
Constructs a ResNet-152 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/backbone/resnet.py:337
Functionresnet18
Constructs a ResNet-18 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/backbone/resnet.py:256
Functionresnet34
Constructs a ResNet-34 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/backbone/resnet.py:284
Functionresnet50
Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
models/backbone/resnet.py:296
Methodset_split
(self, names)
utils/Logger.py:41
Functionsort_coord
(coords)
utils/tools.py:33
Functionvalidate_data
Method validate_data: validates that all files in the results folder are correct (have the correct name contents). Va
cal_rescall/script.py:24
Functionvalidate_lines_in_file
This function validates that all lines of the file calling the Line validation function for each line
cal_rescall/rrc_evaluation_funcs.py:120
Methodweights_init
(self, m)
models/head/seg_detector.py:57
Functionworker_init_fn
(worker_id)
train.py:52
Functionworker_init_fn
(worker_id)
pruned/train_fintune.py:49
← previous101–187 of 187, ranked by callers