Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ShuoYang-1998/Few_Shot_Distribution_Calibration
/ functions
Functions
50 in github.com/ShuoYang-1998/Few_Shot_Distribution_Calibration
⨍
Functions
50
◇
Types & classes
14
↓ 4 callers
Function
GenerateRun
(iRun, cfg, regenRState=False, generate=True)
FSLTask.py:76
↓ 4 callers
Function
mixup_data
Compute the mixup data. Return mixed inputs, pairs of targets, and lambda
wrn_model.py:100
↓ 3 callers
Method
__init__
(self, in_planes, out_planes, stride, dropRate=0.0)
wrn_model.py:20
↓ 2 callers
Function
extract_feature
(val_loader, model, checkpoint_dir, tag='last',set='base')
save_plk.py:46
↓ 2 callers
Method
get_composed_transform
(self, aug = False)
data/datamgr.py:35
↓ 2 callers
Method
get_data_loader
(self, data_file, aug)
data/datamgr.py:47
↓ 2 callers
Function
get_resume_file
(checkpoint_dir)
io_utils.py:44
↓ 2 callers
Function
setRandomStates
(cfg)
FSLTask.py:104
↓ 1 callers
Function
GenerateRunSet
(start=None, end=None, cfg=None)
FSLTask.py:125
↓ 1 callers
Method
__init__
(self, image_size, n_way, n_support, n_query, n_eposide =100)
data/datamgr.py:66
↓ 1 callers
Function
_load_pickle
(file)
FSLTask.py:22
↓ 1 callers
Method
_make_layer
(self, block, in_planes, out_planes, nb_layers, stride, dropRate)
wrn_model.py:73
↓ 1 callers
Function
distribution_calibration
(query, base_means, base_cov, k,alpha=0.21)
evaluate_DC.py:10
↓ 1 callers
Function
loadDataSet
(dsname)
FSLTask.py:42
↓ 1 callers
Function
load_pickle
(file)
save_plk.py:42
↓ 1 callers
Function
parse_args
(script)
io_utils.py:12
↓ 1 callers
Method
parse_transform
(self, transform_type)
data/datamgr.py:19
↓ 1 callers
Function
save_pickle
(file, data)
save_plk.py:38
Function
ClassesInRun
(iRun, cfg)
FSLTask.py:96
Method
__call__
(self, img)
data/additional_transforms.py:20
Method
__getitem__
(self,i)
data/dataset.py:18
Method
__getitem__
(self,i)
data/dataset.py:52
Method
__getitem__
(self,i)
data/dataset.py:65
Method
__init__
(self, module)
save_plk.py:32
Method
__init__
(self, indim, outdim)
wrn_model.py:46
Method
__init__
(self, nb_layers, in_planes, out_planes, block, stride, dropRate=0.0)
wrn_model.py:70
Method
__init__
(self, depth=28, widen_factor=10, num_classes= 200 , loss_type = 'dist', per_img_std = False, stride = 1 )
wrn_model.py:115
Method
__init__
(self, transformdict)
data/additional_transforms.py:16
Method
__init__
(self, data_file, transform, target_transform=identity)
data/dataset.py:11
Method
__init__
(self, data_file, batch_size, transform)
data/dataset.py:30
Method
__init__
(self, sub_meta, cl, transform=transforms.ToTensor(), target_transform=identity)
data/dataset.py:59
Method
__init__
(self, n_classes, n_way, n_episodes)
data/dataset.py:77
Method
__init__
(self, image_size, normalize_param = dict(mean= [0.485, 0.456, 0.406] , std=[0.229, 0.224
data/datamgr.py:12
Method
__init__
(self, image_size, batch_size)
data/datamgr.py:52
Method
__iter__
(self)
data/dataset.py:85
Method
__len__
(self)
data/dataset.py:25
Method
__len__
(self)
data/dataset.py:55
Method
__len__
(self)
data/dataset.py:73
Method
__len__
(self)
data/dataset.py:82
Method
forward
(self, x)
save_plk.py:35
Method
forward
(self, x)
wrn_model.py:34
Method
forward
(self, x)
wrn_model.py:58
Method
forward
(self, x)
wrn_model.py:78
Method
forward
(self, x, target= None, mixup=False, mixup_hidden=True, mixup_alpha=None , lam = 0.4)
wrn_model.py:155
Function
get_assigned_file
(checkpoint_dir, num)
io_utils.py:39
Function
get_best_file
(checkpoint_dir)
io_utils.py:56
Method
get_data_loader
(self, data_file, aug)
data/datamgr.py:57
Method
get_data_loader
(self, data_file, aug)
data/datamgr.py:75
Function
to_one_hot
(inp,num_classes)
wrn_model.py:82
Function
wrn28_10
(num_classes=200 , loss_type = 'dist')
wrn_model.py:208