Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/chuchienshu/ultra-thin-PRM
/ functions
Functions
57 in github.com/chuchienshu/ultra-thin-PRM
⨍
Functions
57
◇
Types & classes
8
↓ 2 callers
Method
backward
(ctx, grad_output)
prm/functions/peak_backprop.py:16
↓ 2 callers
Function
image_transform
Image transforms.
datasets.py:12
↓ 2 callers
Function
peak_stimulation
(input, return_aggregation=True, win_size=3, peak_filter=None)
prm/functions/peak_stimulation.py:51
↓ 1 callers
Method
_patch
(self)
prm/modules/peak_response_mapping.py:58
↓ 1 callers
Method
_read_annotations
(self, split)
datasets.py:173
↓ 1 callers
Method
_recover
(self)
prm/modules/peak_response_mapping.py:64
↓ 1 callers
Function
color_palette
(N)
prm/prm.py:65
↓ 1 callers
Function
fc_resnet50
FC ResNet50.
model.py:77
↓ 1 callers
Function
fetch_voc
Return data loader list.
datasets.py:108
↓ 1 callers
Function
finetune
Fintune.
model.py:20
↓ 1 callers
Method
inference
(self, input_var, raw_img, epoch=0, proposals=None)
solver.py:133
↓ 1 callers
Method
instance_nms
(self, instance_list, threshold=0.3, merge_peak_response=True)
prm/modules/peak_response_mapping.py:69
↓ 1 callers
Method
instance_seg
(self, class_response_maps, peak_list, peak_response_maps, retrieval_cfg)
prm/modules/peak_response_mapping.py:92
↓ 1 callers
Function
main
(args)
main.py:14
↓ 1 callers
Function
pascal_voc_classification
PASCAL VOC dataset.
datasets.py:207
↓ 1 callers
Function
pascal_voc_object_categories
PASCAL VOC dataset class names.
datasets.py:135
↓ 1 callers
Function
peak_response_mapping
Peak Response Mapping.
prm/prm.py:21
↓ 1 callers
Function
prm_visualize
Prediction visualization.
prm/prm.py:41
↓ 1 callers
Method
restore_model
Restore the trained generator and discriminator.
solver.py:47
↓ 1 callers
Function
rgb2hsv
(r, g, b)
prm/prm.py:49
↓ 1 callers
Method
save_checkpoint
(self,state, path, prefix,epoch, filename='checkpoint.pth.tar')
solver.py:66
↓ 1 callers
Function
sgd_optimizer
SGD optimizer.
optims.py:5
↓ 1 callers
Method
train
(self, train_data_loader, train_logger, val_data_loader = None, val_logger = None,resume_iters=0 )
solver.py:72
↓ 1 callers
Method
train
(self, mode=True)
prm/modules/peak_response_mapping.py:251
Method
__getitem__
(self, index)
datasets.py:192
Method
__init__
Initialize configurations.
solver.py:19
Method
__init__
(self, data_dir, dataset, split, classes, transform=None, target_transform=None)
datasets.py:160
Method
__init__
(self, model, num_classes)
model.py:50
Method
__init__
(self, *args, **kargs)
prm/modules/peak_response_mapping.py:15
Method
__init__
(self, model, num_classes)
prm/models/fc_resnet.py:8
Method
__len__
(self)
datasets.py:203
Method
_max_filter
(input)
prm/modules/peak_response_mapping.py:53
Method
_mean_filter
(input)
prm/modules/peak_response_mapping.py:47
Method
_median_filter
(input)
prm/modules/peak_response_mapping.py:41
Function
adadelta_optimizer
Adadelta optimizer.
optims.py:18
Method
backward
(ctx, grad_peak_list, grad_output)
prm/functions/peak_stimulation.py:44
Method
backward
(ctx, grad_output)
prm/functions/peak_backprop.py:28
Method
count_parameters
(self, model)
solver.py:63
Function
cross_entropy_loss
Cross entropy loss.
losses.py:6
Function
fc_resnet50
FC ResNet50.
prm/prm.py:14
Function
fetch_data
Return data loader list.
datasets.py:57
Method
forward
(self, x)
model.py:69
Method
forward
(self, input, class_threshold=0, peak_threshold=30, retrieval_cfg=None)
prm/modules/peak_response_mapping.py:173
Method
forward
(ctx, input, return_aggregation, win_size, peak_filter)
prm/functions/peak_stimulation.py:9
Method
forward
(ctx, input, offset)
prm/functions/peak_backprop.py:11
Method
forward
(ctx, input, norm_factor)
prm/functions/peak_backprop.py:23
Method
forward
(self, x)
prm/models/fc_resnet.py:27
Method
inference
(self)
prm/modules/peak_response_mapping.py:258
Method
iou_filter
(x)
prm/modules/peak_response_mapping.py:76
Function
multilabel_soft_margin_loss
Multilabel soft margin loss.
losses.py:19
Function
peak_response_mapping
Peak Response Mapping.
model.py:84
Function
pr_conv2d
(self, input)
prm/functions/peak_backprop.py:37
Method
print_network
Print out the network information.
solver.py:38
Function
rle_decode
Decode a Run-Length Encoding (RLE).
utils.py:19
Function
rle_encode
Perform Run-Length Encoding (RLE) on a binary mask.
utils.py:7
Method
update_lr
(self, lr)
solver.py:59
Method
validation
(self, data_loader,test_logger,inference_epoch=0)
solver.py:209