MCPcopy Create free account

hub / github.com/LiheYoung/MiningFSS / functions

Functions35 in github.com/LiheYoung/MiningFSS

↓ 6 callersFunctionconv3x3
(in_planes, out_planes, stride=1, groups=1, dilation=1)
model/resnet.py:9
↓ 5 callersFunction_resnet
(arch, block, layers, pretrained, **kwargs)
model/resnet.py:199
↓ 3 callersMethod_make_layer
:param last_relu: in metric learning paradigm, the final relu is removed (last_relu = False)
model/resnet.py:158
↓ 3 callersFunctionconv1x1
(in_planes, out_planes, stride=1)
model/resnet.py:14
↓ 3 callersFunctionset_seed
(seed)
util/utils.py:11
↓ 2 callersMethod__init__
(self, block, layers, zero_init_residual=False, groups=1, width_per_group=64, replace_stride_
model/resnet.py:105
↓ 2 callersMethodbase_forward
(self, x)
model/resnet.py:186
↓ 2 callersFunctioncrop
(img, mask, size)
dataset/transform.py:8
↓ 2 callersFunctionevaluate
(model, dataloader, args)
main.py:73
↓ 2 callersMethodevaluate
(self)
util/utils.py:33
↓ 2 callersFunctionhflip
(img, mask)
dataset/transform.py:26
↓ 2 callersMethodmasked_average_pooling
(self, feature, mask)
model/matching.py:44
↓ 2 callersFunctionnormalize
:param img: PIL image :param mask: PIL image, corresponding mask :return: normalized torch tensor of image and mask
dataset/transform.py:33
↓ 1 callersMethod_fast_hist
(self, label_pred, label_true)
util/utils.py:22
↓ 1 callersMethod_filter_ids
(self)
dataset/fewshot.py:100
↓ 1 callersMethod_map_cls_to_cls
(self)
dataset/fewshot.py:118
↓ 1 callersMethodadd_batch
(self, predictions, gts)
util/utils.py:29
↓ 1 callersFunctioncount_params
(model)
util/utils.py:6
↓ 1 callersFunctionmain
()
main.py:99
↓ 1 callersFunctionparse_args
()
main.py:16
Method__getitem__
(self, item)
dataset/fewshot.py:48
Method__init__
(self, dataset, root, size, mode, fold, shot, episode)
dataset/fewshot.py:19
Method__init__
(self, backbone)
model/matching.py:9
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
model/resnet.py:21
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
model/resnet.py:61
Method__init__
(self, num_classes)
util/utils.py:18
Method__len__
(self)
dataset/fewshot.py:95
Methodforward
(self, img_s_list, mask_s_list, img_q)
model/matching.py:13
Methodforward
(self, x)
model/resnet.py:39
Methodforward
(self, x)
model/resnet.py:79
Functionresnet101
(pretrained=False)
model/resnet.py:220
Functionresnet152
(pretrained=False)
model/resnet.py:225
Functionresnet18
(pretrained=False)
model/resnet.py:207
Functionresnet34
(pretrained=False)
model/resnet.py:211
Functionresnet50
(pretrained=False)
model/resnet.py:215