MCPcopy Create free account

hub / github.com/Haochen-Wang409/U2PL / functions

Functions166 in github.com/Haochen-Wang409/U2PL

⨍ Functions 166◇ Types & classes 33
Method__init__
(self, mean, std=None)
u2pl/dataset/augmentation.py:96
Method__init__
(self, size)
u2pl/dataset/augmentation.py:120
Method__init__
(self, size=2048)
u2pl/dataset/augmentation.py:137
Method__init__
(self, scale, aspect_ratio=None)
u2pl/dataset/augmentation.py:164
Method__init__
(self, rotate, ignore_label=255)
u2pl/dataset/augmentation.py:274
Method__init__
(self, radius=2)
u2pl/dataset/augmentation.py:316
Method__init__
(self, radius)
u2pl/dataset/augmentation.py:326
Method__init__
(self, n_holes, length)
u2pl/dataset/augmentation.py:356
Method__init__
( self, prop_range, n_holes=1, random_aspect_ratio=True, within_bounds=True )
u2pl/dataset/augmentation.py:405
Method__init__
(self, d_list, **kwargs)
u2pl/dataset/base.py:8
Method__init__
( self, dataset, total_iter, batch_size, world_size=None, rank=None, last_iter=-1 )
u2pl/dataset/sampler.py:13
Method__init__
( self, data_root, data_list, trs_form, seed=0, n_sup=10582, split="val" )
u2pl/dataset/pascal_voc.py:18
Method__init__
(self, data_root, data_list, trs_form, seed, n_sup, split="val")
u2pl/dataset/cityscapes.py:19
Method__init__
( self, in_planes, num_classes=19, inner_planes=256, sync_bn=False,
u2pl/models/decoder.py:9
Method__init__
( self, in_planes, num_classes=19, inner_planes=256, sync_bn=False,
u2pl/models/decoder.py:46
Method__init__
(self, net_cfg)
u2pl/models/model_helper.py:10
Method__init__
( self, in_planes, inner_planes=256, sync_bn=False, dilations=(12, 24, 36) )
u2pl/models/base.py:17
Method__init__
( self, inplanes, planes, stride=1, downsample=None, groups=1,
u2pl/models/resnet.py:47
Method__init__
( self, inplanes, planes, stride=1, downsample=None, groups=1,
u2pl/models/resnet.py:96
Method__iter__
(self)
u2pl/utils/dist_helper.py:77
Method__iter__
(self)
u2pl/dataset/sampler.py:33
Method__len__
(self)
u2pl/utils/dist_helper.py:105
Method__len__
(self)
u2pl/dataset/base.py:48
Method__len__
(self)
u2pl/dataset/sampler.py:64
Method__len__
(self)
u2pl/dataset/pascal_voc.py:44
Method__len__
(self)
u2pl/dataset/cityscapes.py:43
Functionaccuracy
(output, target, topk=(1,))
u2pl/utils/utils.py:511
Functioncal_category_confidence
( preds_student_sup, preds_student_unsup, gt, preds_teacher_unsup, num_classes )
u2pl/utils/utils.py:418
Functioncal_pixel_num
(pred_map)
u2pl/utils/utils.py:97
Functioncheck_mkdir
(dir_name)
u2pl/utils/utils.py:534
Functioncompute_rce_loss
(predict, target)
u2pl/utils/loss_helper.py:10
Functiondynamic_copy_paste
(images_sup, labels_sup, query_cat)
u2pl/utils/utils.py:389
Methodforward
(self, preds, target)
u2pl/utils/loss_helper.py:295
Methodforward
(self, preds, target)
u2pl/utils/loss_helper.py:339
Methodforward
Args: predict:(n, c, h, w) target:(n, h, w) weight (Tensor, optional): a manual rescaling weight given to
u2pl/utils/loss_helper.py:436
Methodforward
(self, pred, target)
u2pl/utils/loss_helper.py:502
Methodforward
(self, input)
u2pl/dataset/augmentation.py:336
Methodforward
(self, x)
u2pl/models/decoder.py:39
Methodforward
(self, x)
u2pl/models/decoder.py:108
Methodforward
(self, x)
u2pl/models/decoder.py:140
Methodforward
(self, x)
u2pl/models/model_helper.py:49
Methodforward
(self, x)
u2pl/models/base.py:90
Methodforward
(self, x)
u2pl/models/resnet.py:74
Methodforward
(self, x)
u2pl/models/resnet.py:120
Methodforward
(self, x)
u2pl/models/resnet.py:279
Functiongather_together
(data)
u2pl/utils/dist_helper.py:49
Functiongenerate_cutmix_mask
( pred, sample_cat, area_thresh=0.0001, no_pad=False, no_slim=False )
u2pl/utils/utils.py:290
Methodget_auxplanes
(self)
u2pl/models/resnet.py:229
Methodget_outplanes
(self)
u2pl/models/resnet.py:226
Functionget_palette
Returns the color map for visualizing the segmentation mask. Args: num_cls: Number of classes Returns: The color map
u2pl/utils/utils.py:544
Functionget_world_size
()
u2pl/utils/utils.py:62
Functionignore_state_head
(state_dict)
u2pl/utils/utils.py:502
Functioninit_cutmix_bank
(cutmix_bank, crop_size)
u2pl/utils/utils.py:352
Functionis_main_process
()
u2pl/utils/utils.py:78
Functionmap_func
(storage, location)
u2pl/utils/utils.py:586
Functionpadding_bbox
(rectangles, size)
u2pl/utils/utils.py:209
Functionpadding_bbox_old
(rectangles, size)
u2pl/utils/utils.py:127
Functionresnet101
Constructs a ResNet-101 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
u2pl/models/resnet.py:361
Functionresnet152
Constructs a ResNet-152 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
u2pl/models/resnet.py:383
Functionresnet18
Constructs a ResNet-18 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
u2pl/models/resnet.py:295
Functionresnet34
Constructs a ResNet-34 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
u2pl/models/resnet.py:317
Functionresnet50
Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
u2pl/models/resnet.py:339
Functionsample_from_bank
(cutmix_bank, conf, smooth=False)
u2pl/utils/utils.py:274
Functionsynchronize
Helper function to synchronize (barrier) among all processes when using distributed training
u2pl/utils/utils.py:82
Functionupdate_cutmix_bank
( cutmix_bank, preds_teacher_unsup, img_id, sample_id, area_thresh=0.0001 )
u2pl/utils/utils.py:309
Functionupdate_cutmix_mask
(pred_map, num_classes)
u2pl/utils/utils.py:339
← previous101–166 of 166, ranked by callers