MCPcopy Create free account

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

Functions166 in github.com/Haochen-Wang409/U2PL

⨍ Functions 166◇ Types & classes 33
↓ 27 callersMethodupdate
(self, val, num=1)
u2pl/utils/utils.py:454
↓ 6 callersFunctionconv3x3
3x3 convolution with padding
u2pl/models/resnet.py:25
↓ 6 callersFunctionload_state
(path, model, optimizer=None, key="state_dict")
u2pl/utils/utils.py:583
↓ 5 callersFunctionget_syncbn
()
u2pl/models/base.py:6
↓ 4 callersMethod_make_layer
( self, block, planes, blocks, stride=1, dilate=False, multi_grid=False )
u2pl/models/resnet.py:232
↓ 4 callersFunctionget_optimizer
Get the optimizer
u2pl/utils/lr_helper.py:12
↓ 4 callersMethodimg_loader
(self, path, mode)
u2pl/dataset/base.py:43
↓ 4 callersMethodstep
(self)
u2pl/utils/lr_helper.py:78
↓ 3 callersMethod__init__
(self, aux_weight, ignore_index=255, use_weight=False)
u2pl/utils/loss_helper.py:259
↓ 3 callersFunctionbuild_cityloader
(split, all_cfg, seed=0)
u2pl/dataset/cityscapes.py:80
↓ 3 callersFunctionbuild_transfrom
(cfg)
u2pl/dataset/pascal_voc.py:48
↓ 3 callersFunctionbuild_transfrom
(cfg)
u2pl/dataset/cityscapes.py:47
↓ 3 callersFunctionbuild_vocloader
(split, all_cfg, seed=0)
u2pl/dataset/pascal_voc.py:74
↓ 3 callersFunctionconv1x1
1x1 convolution
u2pl/models/resnet.py:39
↓ 3 callersMethodget_outplanes
(self)
u2pl/models/base.py:87
↓ 3 callersFunctionintersectionAndUnion
(output, target, K, ignore_index=255)
u2pl/utils/utils.py:568
↓ 2 callersMethod__init__
(self, in_planes, num_classes=19, sync_bn=False)
u2pl/models/decoder.py:128
↓ 2 callersMethod__init__
( self, block, layers, zero_init_residual=False, groups=1, wid
u2pl/models/resnet.py:144
↓ 2 callersMethod_build_module
(self, mtype, kwargs)
u2pl/models/model_helper.py:43
↓ 2 callersFunctioncheck_makedirs
(dir_name)
u2pl/utils/utils.py:539
↓ 2 callersFunctioncolorize
(mask, colormap)
u2pl/utils/utils.py:526
↓ 2 callersFunctioncompute_contra_memobank_loss
( rep, label_l, label_u, prob_l, prob_u, low_mask, high_mask, cfg, memoban
u2pl/utils/loss_helper.py:51
↓ 2 callersFunctionconvert_state_dict
(state_dict)
u2pl/utils/utils.py:494
↓ 2 callersFunctiongenerate_cutmix
(pred, cat, area_thresh, no_pad=False, no_slim=False)
u2pl/utils/utils.py:244
↓ 2 callersFunctiongenerate_cutout_mask
(img_size, ratio=2)
u2pl/dataset/augmentation.py:471
↓ 2 callersFunctionget_criterion
(cfg)
u2pl/utils/loss_helper.py:238
↓ 2 callersFunctionget_loader
(cfg, seed=0)
u2pl/dataset/builder.py:9
↓ 2 callersMethodget_lr
(self)
u2pl/utils/lr_helper.py:108
↓ 2 callersFunctionget_scheduler
(cfg_trainer, len_data, optimizer, start_epoch=0, use_iteration=False)
u2pl/utils/lr_helper.py:30
↓ 2 callersFunctioninit_cutmix
(crop_size)
u2pl/utils/utils.py:106
↓ 2 callersFunctioninit_log
(name, level=logging.INFO)
u2pl/utils/utils.py:474
↓ 2 callersFunctionlabel_onehot
(inputs, num_segments)
u2pl/utils/utils.py:50
↓ 2 callersFunctionnet_process
(model, image)
eval.py:159
↓ 2 callersFunctionset_random_seed
Set random seed.
u2pl/utils/utils.py:378
↓ 2 callersFunctionsetup_distributed
AdaHessian Optimizer Lifted from https://github.com/BIGBALLON/distribuuuu/blob/master/distribuuuu/utils.py Originally licensed MIT, Copyright
u2pl/utils/dist_helper.py:13
↓ 2 callersFunctionvalidate
( model, data_loader, epoch, logger, )
train_semi.py:595
↓ 1 callersMethod__init__
(self, size, crop_type="center", ignore_label=255)
u2pl/dataset/augmentation.py:221
↓ 1 callersMethod_build_decoder
(self, dec_cfg)
u2pl/models/model_helper.py:32
↓ 1 callersMethod_build_encoder
(self, enc_cfg)
u2pl/models/model_helper.py:27
↓ 1 callersMethod_step
(self)
u2pl/utils/lr_helper.py:83
↓ 1 callersFunctionbuild_city_semi_loader
(split, all_cfg, seed=0)
u2pl/dataset/cityscapes.py:108
↓ 1 callersFunctionbuild_voc_semi_loader
(split, all_cfg, seed=0)
u2pl/dataset/pascal_voc.py:101
↓ 1 callersFunctioncolorful
(mask, colormap)
infer.py:137
↓ 1 callersFunctioncompute_unsupervised_loss
(predict, target, percent, pred_teacher)
u2pl/utils/loss_helper.py:30
↓ 1 callersFunctioncreate_cityscapes_label_colormap
Creates a label colormap used in CityScapes segmentation benchmark. Returns: A colormap for visualizing segmentation results.
u2pl/utils/utils.py:639
↓ 1 callersFunctioncreate_pascal_label_colormap
Creates a label colormap used in Pascal segmentation benchmark. Returns: A colormap for visualizing segmentation results.
infer.py:145
↓ 1 callersFunctioncreate_pascal_label_colormap
Creates a label colormap used in Pascal segmentation benchmark. Returns: A colormap for visualizing segmentation results.
u2pl/utils/utils.py:668
↓ 1 callersFunctiondequeue_and_enqueue
(keys, queue, queue_ptr, queue_size)
u2pl/utils/utils.py:28
↓ 1 callersMethodfind_threshold
(self, np_predict, np_target)
u2pl/utils/loss_helper.py:372
↓ 1 callersFunctiongather_together
(data)
u2pl/utils/utils.py:17
↓ 1 callersMethodgen_new_list
(self)
u2pl/utils/dist_helper.py:86
↓ 1 callersMethodgen_new_list
(self)
u2pl/dataset/sampler.py:42
↓ 1 callersFunctiongenerate_class_mask
(pseudo_labels)
u2pl/dataset/augmentation.py:488
↓ 1 callersMethodgenerate_new_target
(self, predict, target)
u2pl/utils/loss_helper.py:404
↓ 1 callersFunctiongenerate_unsup_data
(data, target, logits, mode="cutout")
u2pl/dataset/augmentation.py:498
↓ 1 callersFunctionget_logger
()
eval.py:61
↓ 1 callersFunctionget_logger
()
infer.py:42
↓ 1 callersFunctionget_parser
()
eval.py:28
↓ 1 callersFunctionget_parser
()
infer.py:27
↓ 1 callersFunctionget_rank
()
u2pl/utils/utils.py:70
↓ 1 callersFunctionmain
()
train_sup.py:42
↓ 1 callersFunctionmain
()
eval.py:72
↓ 1 callersFunctionmain
()
train_semi.py:47
↓ 1 callersFunctionmain
()
infer.py:53
↓ 1 callersFunctionnet_process
(model, image)
infer.py:177
↓ 1 callersFunctionpadding_bbox_new
(rectangles, size)
u2pl/utils/utils.py:153
↓ 1 callersMethodparse_input_list
(self, d_list, max_sample=-1, start_idx=-1, end_idx=-1)
u2pl/dataset/base.py:12
↓ 1 callersMethodreset
(self)
u2pl/utils/utils.py:445
↓ 1 callersFunctionscale_crop_process
(model, image, classes, crop_h, crop_w, h, w, stride_rate=2 / 3)
eval.py:184
↓ 1 callersFunctionscale_whole_process
(model, image, h, w)
eval.py:226
↓ 1 callersFunctionsliming_bbox
(rectangles, size)
u2pl/utils/utils.py:175
↓ 1 callersFunctiontrain
( model, optimizer, lr_scheduler, criterion, data_loader, epoch, tb_logger, )
train_sup.py:177
↓ 1 callersFunctiontrain
( model, model_teacher, optimizer, lr_scheduler, sup_loss_fn, loader_l, loader_u,
train_semi.py:234
↓ 1 callersMethodupdate_lr
(self)
u2pl/utils/lr_helper.py:111
↓ 1 callersFunctionvaliadte_whole
( model, classes, data_list, mean, std, scales, gray_folder, color_folder )
eval.py:313
↓ 1 callersFunctionvalidate
( model, data_loader, epoch, )
train_sup.py:254
↓ 1 callersFunctionvalidate_city
( model, classes, data_list, mean, std, base_size, crop_h, crop_w, scales,
eval.py:235
↓ 1 callersMethodweight_init
(self)
u2pl/dataset/augmentation.py:340
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:31
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:52
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:104
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:124
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:141
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:190
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:247
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:285
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:300
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:308
Method__call__
(self, image, label)
u2pl/dataset/augmentation.py:319
Method__call__
Args: img (Tensor): Tensor image of size (C, H, W). Returns: Tensor: Image with n_holes of dimension length x
u2pl/dataset/augmentation.py:360
Method__call__
Args: img (Tensor): Tensor image of size (C, H, W). Returns: Tensor: Image with n_holes of dimension length x
u2pl/dataset/augmentation.py:414
Method__getitem__
(self, index)
u2pl/dataset/pascal_voc.py:35
Method__getitem__
(self, index)
u2pl/dataset/cityscapes.py:34
Method__init__
( self, dataset, total_iter, batch_size, world_size=None, rank=None, last_iter=-1 )
u2pl/utils/dist_helper.py:57
Method__init__
( self, aux_weight, thresh=0.7, min_kept=100000, ignore_index=255,
u2pl/utils/loss_helper.py:324
Method__init__
(self, ignore_label=255, thresh=0.7, min_kept=100000, factor=8)
u2pl/utils/loss_helper.py:364
Method__init__
( self, ignore_index=255, thresh=0.7, min_kept=256, use_weight=False, reduce=False )
u2pl/utils/loss_helper.py:456
Method__init__
(self, length=0)
u2pl/utils/utils.py:441
Method__init__
(self, mode, lr_args, data_size, optimizer, num_epochs, start_epochs)
u2pl/utils/lr_helper.py:43
Method__init__
(self, segtransforms)
u2pl/dataset/augmentation.py:28
next →1–100 of 166, ranked by callers