MCPcopy Create free account

hub / github.com/Shank2358/TS-Conv / functions

Functions408 in github.com/Shank2358/TS-Conv

Methodforward
(self, input, rois, offset)
lib/DCNv2/dcn_v2_amp.py:571
Methodforward
(self, input, rois)
lib/DCNv2/dcn_v2_amp.py:630
Methodforward
(self, x)
model/TSConv.py:34
Methodforward
(self, x)
model/backbones/darknet53.py:55
Methodforward
(self, x: Tensor)
model/backbones/model_resnet.py:87
Methodforward
(self, x: Tensor)
model/backbones/model_resnet.py:141
Methodforward
(self, x: Tensor)
model/backbones/model_resnet.py:280
Methodforward
(self, x)
model/backbones/mobilenetv2.py:77
Methodforward
(self, x)
model/backbones/mobilenetv2.py:114
Methodforward
(self, x)
model/backbones/mobilenetv2.py:147
Methodforward
(self, x)
model/backbones/mobilenetv2.py:177
Methodforward
(self, x)
model/backbones/resnet.py:36
Methodforward
(self, x)
model/backbones/resnet.py:74
Methodforward
(self, input1)
model/head/head.py:20
Methodforward
(self, out1, loc, cls, offsets_loc, offsets_cls, mask_loc, mask_cls, w_c8)
model/head/head.py:199
Methodforward
(self,x)
model/layers/attention_blocks.py:5
Methodforward
(self, x)
model/layers/attention_blocks.py:26
Methodforward
(self, x)
model/layers/attention_blocks.py:44
Methodforward
(self, x)
model/layers/attention_blocks.py:129
Methodforward
(self, x)
model/layers/attention_blocks.py:168
Methodforward
(self, x)
model/layers/multiscale_fusion_blocks.py:14
Methodforward
(self, x)
model/layers/multiscale_fusion_blocks.py:28
Methodforward
(self, x)
model/layers/multiscale_fusion_blocks.py:44
Methodforward
(self, x_level_0, x_level_1, x_level_2)
model/layers/multiscale_fusion_blocks.py:78
Methodforward
(self, input, wh_pred)
model/layers/multiscale_fusion_blocks.py:121
Methodforward
(self, x1, x2)
model/layers/multiscale_fusion_blocks.py:137
Methodforward
(self, x)
model/layers/np_attention_blocks.py:47
Methodforward
(self, x)
model/layers/np_attention_blocks.py:77
Methodforward
(self, x)
model/layers/np_attention_blocks.py:93
Methodforward
(self, x, mask)
model/layers/np_attention_blocks.py:136
Methodforward
(self, x)
model/layers/np_attention_blocks.py:200
Methodforward
(self, x)
model/layers/np_attention_blocks.py:263
Methodforward
(self, x)
model/layers/conv_blocks.py:13
Methodforward
(self, x)
model/layers/conv_blocks.py:28
Methodforward
(self, x)
model/layers/conv_blocks.py:42
Methodforward
(self, x)
model/layers/conv_blocks.py:72
Methodforward
(x)
model/layers/activations.py:7
Methodforward
(x)
model/layers/activations.py:12
Methodforward
(ctx, x)
model/layers/activations.py:19
Methodforward
(self, x)
model/layers/activations.py:27
Methodforward
(x)
model/layers/activations.py:32
Methodforward
(ctx, x)
model/layers/activations.py:38
Methodforward
(self, x)
model/layers/activations.py:47
Methodforward
(self, x)
model/layers/activations.py:56
Methodforward
(self, x)
model/layers/msr_blocks.py:22
Methodforward
(self, x)
model/layers/msr_blocks.py:49
Methodforward
(self, x)
model/layers/msr_blocks.py:75
Methodforward
(self, x)
model/layers/convolutions.py:54
Methodforward
(self, x)
model/layers/convolutions.py:92
Methodforward
(self, x)
model/layers/convolutions.py:114
Methodforward
(self, x, routing_weight)
model/layers/convolutions.py:148
Methodforward
(self, x)
model/layers/convolutions.py:201
Methodforward
(self, x)
model/neck/neck.py:18
Methodforward
(self, x1, x2)
model/neck/neck.py:26
Methodforward
(self, x0, x1, x2)
model/neck/neck.py:178
Methodforward
(self, input, target)
model/loss/loss.py:14
Methodforward
(self, input, target)
model/loss/loss.py:22
Methodforward
(self, p1, p1_d, p2, p2_d, label_sbbox, label_mbbox, label_lbbox, epoch, i)
model/loss/loss.py:42
Functionget_coco_label_names
COCO label names and correspondence between the model's class index and COCO class index. Returns: coco_label_names (tuple of str) :
utils/utils_coco.py:239
Functionget_sha
()
utils/mics.py:248
Methodglobal_avg
(self)
utils/mics.py:68
Functioninit_distributed_mode
(args)
utils/mics.py:406
Functioninit_seeds
(seed=0)
utils/gpu.py:9
Functioninit_seeds
(seed=0)
utils/utils_basic.py:12
Functioninterpolate
Equivalent to nn.functional.interpolate, but with support for empty batch sizes. This will eventually be supported natively by PyTorch, and t
utils/mics.py:450
Functioniou_xywh_numpy
(boxes1, boxes2)
utils/utils_basic.py:39
Functioniou_xywh_torch
:param boxes1: boxes1和boxes2的shape可以不相同,但是需要满足广播机制,且需要是Tensor :param boxes2: 且需要保证最后一维为坐标维,以及坐标的存储结构为(x, y, w, h) :return: 返回boxes1和boxes
utils/utils_basic.py:237
Functioniou_xyxy_torch
:param boxes1: boxes1和boxes2的shape可以不相同,但是需要满足广播机制,且需要是Tensor :param boxes2: 且需要保证最后一维为坐标维,以及坐标的存储结构为(xmin, ymin, xmax, ymax) :return: 返回
utils/utils_basic.py:217
Functionlabel2box
(labels, info_img, maxsize, lrflip)
utils/utils_coco.py:126
Methodload_resnet101_weights
(self, weight_file='/home/hzc/v2/weight/resnet101-cd907fc2.pth')
model/TSConv.py:75
Methodlog_every
(self, iterable, print_freq, header=None)
utils/mics.py:193
Methodmedian
(self)
utils/mics.py:58
Functionnms
:param bboxes: 假设有N个bbox的score大于score_threshold,那么bboxes的shape为(N, 6),存储格式为(xmin, ymin, xmax, ymax, score, class) 其中(xmin, ymin, xmax, ym
utils/utils_basic.py:397
Methodnon_max_suppression_4points1
Performs Rotate-Non-Maximum Suppression (RNMS) on inference results; @param prediction: size=(batch_size, num, [xywh,score,num_classe
evalR/evaluatorTSplot.py:311
Functionparse_rec
Parse a PASCAL VOC xml file
evalR/voc_eval.py:8
Functionpolygen_iou_xy4_numpy1
(boxes1, boxes2)
utils/utils_basic.py:142
Functionpolygen_iou_xy4_torch
(boxes1, boxes2)
utils/utils_basic.py:161
Functionpolygen_iou_xy4_torch1
(boxes1, boxes2)
utils/utils_basic.py:181
Functionpostprocess
(prediction, num_classes, conf_thre=0.7, nms_thre=0.45)
utils/utils_coco.py:54
Functionpreprocess
(img, imgsize, jitter, random_placing=False)
utils/utils_coco.py:154
Functionrandom_distort
perform random distortion in the HSV color space. Args: img (numpy.ndarray): input image whose shape is :math:`(H, W, C)`.
utils/utils_coco.py:204
Functionreduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
utils/mics.py:131
Functionresnet101
r"""ResNet-101 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_. Args: pretrained
model/backbones/model_resnet.py:332
Functionresnet152
r"""ResNet-152 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_. Args: pretrained
model/backbones/model_resnet.py:343
Functionresnet18
r"""ResNet-18 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_. Args: pretrained (
model/backbones/model_resnet.py:299
Functionresnet34
r"""ResNet-34 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_. Args: pretrained (
model/backbones/model_resnet.py:310
Functionresnet50
r"""ResNet-50 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_. Args: pretrained (
model/backbones/model_resnet.py:321
Functionresnext101_32x8d
r"""ResNeXt-101 32x8d model from `"Aggregated Residual Transformation for Deep Neural Networks" <https://arxiv.org/pdf/1611.05431.pdf>`_. Arg
model/backbones/model_resnet.py:367
Functionresnext50_32x4d
r"""ResNeXt-50 32x4d model from `"Aggregated Residual Transformation for Deep Neural Networks" <https://arxiv.org/pdf/1611.05431.pdf>`_. Args
model/backbones/model_resnet.py:354
Functionsave_on_master
(*args, **kwargs)
utils/mics.py:401
Functionselect_device
(id, force_cpu=False)
utils/gpu.py:30
Functionselect_device_v5
(device='', batch_size=None)
utils/gpu.py:52
Functionset_amp
(amp)
lib/DCNv2/dcn_v2_amp.py:18
Methodsmooth_l1_loss
(self, input, target, beta=1. / 9, size_average=True)
model/loss/loss.py:66
Methodsymbolic
(g, input, offset_mask, weight, bias, stride, padding, dilation, deformable_groups)
lib/DCNv2/dcn_v2_onnx.py:20
Methodsymbolic
( g, input, offset, mask, weight, bias, stride, padding, dilation, deformable_groups )
lib/DCNv2/dcn_v2.py:69
Methodsynchronize
Helper function to synchronize (barrier) among all processes when using distributed training
trainv2.py:242
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
utils/mics.py:44
Methodsynchronize_between_processes
(self)
utils/mics.py:186
Functiontorch_distributed_zero_first
Decorator to make all processes in distributed training wait for each local_master to do something.
utils/gpu.py:19
← previousnext →301–400 of 408, ranked by callers