Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ChunmingHe/WS-SAM
/ functions
Functions
357 in github.com/ChunmingHe/WS-SAM
⨍
Functions
357
◇
Types & classes
99
Method
__init__
(self, distance_func)
losses/kernels/lib_tree_filter/modules/tree_filter.py:10
Method
__len__
(self)
utils/dataloader.py:69
Method
__len__
(self)
utils/data_val.py:219
Method
__len__
(self)
utils/data_val.py:321
Method
__len__
(self)
utils/data_val.py:423
Method
__len__
(self)
utils/data_val.py:538
Method
__len__
(self)
utils/data_val.py:655
Method
__len__
(self)
utils/data_val.py:764
Method
__len__
(self)
utils/data_val.py:828
Function
_conv_filter
convert patch embedding weight from manual patchify + linear proj to conv
lib/VIT/encoder/pvtv2_encoder.py:385
Method
_init_weights
(self, m)
lib/VIT/encoder/pvtv2_encoder.py:30
Method
_init_weights
(self, m)
lib/VIT/encoder/pvtv2_encoder.py:78
Method
_init_weights
(self, m)
lib/VIT/encoder/pvtv2_encoder.py:135
Method
_init_weights
(self, m)
lib/VIT/encoder/pvtv2_encoder.py:176
Method
_init_weights
(self, m)
lib/VIT/encoder/pvtv2_encoder.py:259
Method
_init_weights
(self, m)
lib/VIT/encoder/swin_encoder.py:546
Method
_make_pred_layer
(self, block, dilation_series, padding_series, NoLabels, input_channel)
lib/VIT/CamoFormer.py:74
Function
adjust_lr
(optimizer, init_lr, epoch, decay_rate=0.1, decay_epoch=30)
utils/utils.py:48
Method
backward
(ctx, grad_output)
losses/kernels/lib_tree_filter/functions/refine.py:23
Function
binary_cross_entropy
(pred, label, weight=None, reduc
losses/cross_entropy_loss.py:48
Function
boruvkaMST
The main function for MST using Boruvka's algorithm
losses/kernels/lib_tree_filter/src/mst/boruvka.cpp:20
Function
createGraph
Creates a graph with V vertices and E edges
losses/kernels/lib_tree_filter/src/mst/boruvka.cpp:115
Function
cross_entropy
(pred, label, weight=None, reduction='mean', avg_factor=None)
losses/cross_entropy_loss.py:19
Function
data_split
Usage: randomly spliting dataset :param src_list: :return:
utils/tif2png.py:21
Function
dice_loss
(predict, target)
Train.py:56
Method
extra_repr
(self)
lib/VIT/encoder/swin_encoder.py:149
Method
extra_repr
(self)
lib/VIT/encoder/swin_encoder.py:275
Method
extra_repr
(self)
lib/VIT/encoder/swin_encoder.py:333
Method
extra_repr
(self)
lib/VIT/encoder/swin_encoder.py:401
Method
flops
(self, N)
lib/VIT/encoder/swin_encoder.py:152
Method
flops
(self)
lib/VIT/encoder/swin_encoder.py:279
Method
flops
(self)
lib/VIT/encoder/swin_encoder.py:336
Method
flops
(self)
lib/VIT/encoder/swin_encoder.py:453
Method
flops
(self)
lib/VIT/encoder/swin_encoder.py:596
Method
forward
(self, x)
lib/Network.py:28
Method
forward
(self, x)
lib/Network.py:51
Method
forward
(self, input)
lib/GatedConv.py:26
Method
forward
(self, inputs)
lib/Slot.py:34
Method
forward
(self, inputs, num_slots=None)
lib/Slot.py:113
Method
forward
(self, image)
lib/Slot.py:231
Method
forward
(self, x)
lib/Modules.py:28
Method
forward
(self, x)
lib/Modules.py:44
Method
forward
(self, x)
lib/Modules.py:62
Method
forward
(self, x)
lib/Modules.py:84
Method
forward
(self, x)
lib/Modules.py:139
Method
forward
(self, x)
lib/Modules.py:186
Method
forward
(self, x)
lib/Modules.py:216
Method
forward
(self, x)
lib/Modules.py:228
Method
forward
(self, x)
lib/Modules.py:242
Method
forward
(self, f1, f2)
lib/Modules.py:255
Method
forward
(self, f1, f2, f3, f4)
lib/Modules.py:276
Method
forward
(self, inputs, num_slots=None)
lib/Modules.py:336
Method
forward
(self, x)
lib/Modules.py:397
Method
forward
(self, x)
lib/Modules.py:448
Method
forward
(self, x)
lib/Modules.py:485
Method
forward
(self, feature_map)
lib/Modules.py:499
Method
forward
(self, feature_map)
lib/Modules.py:513
Method
forward
(self, f1, f2, f3, f4)
lib/Modules.py:534
Method
forward
(self, f1, f2, f3, f4)
lib/Modules.py:561
Method
forward
(self, f1, f2, f3, f4)
lib/Modules.py:590
Method
forward
(self, f1, f2, f3, f4)
lib/Modules.py:619
Method
forward
(self, f1, f2, f3, f4)
lib/Modules.py:651
Method
forward
(self, x)
lib/Modules.py:679
Method
forward
(self, x)
lib/Modules.py:704
Method
forward
(self, f4, prior_cam)
lib/Modules.py:721
Method
forward
(self, x, edge)
lib/Modules.py:746
Method
forward
(self, f, f_s, f_e, p_s, p_e)
lib/Modules.py:771
Method
forward
(self, x, prior_cam, pic)
lib/Modules.py:801
Method
forward
(self, f1, f2, f3, f4)
lib/Modules.py:834
Method
forward
(self, f4, prior_cam)
lib/Modules.py:854
Method
forward
(self, f, f_s, f_e, p_s, p_e)
lib/Modules.py:878
Method
forward
(self, x, prior_cam, pic)
lib/Modules.py:905
Method
forward
(self, x, prior_cam, pic)
lib/Modules.py:933
Method
forward
(self, f4, prior_cam)
lib/Modules.py:961
Method
forward
(self, f, f_s, f_e, p_s, p_e)
lib/Modules.py:987
Method
forward
(self, x, prior_cam, pic)
lib/Modules.py:1016
Method
forward
(self, x)
lib/Res2Net_v1b.py:58
Method
forward
(self, x)
lib/Res2Net_v1b.py:147
Method
forward
(self, x)
lib/resnet.py:69
Method
forward
(self, x)
lib/resnet.py:109
Method
forward
(self, x)
lib/resnet.py:227
Method
forward
(self, x, shape=None, name=None)
lib/VIT/CamoFormer.py:77
Method
forward
(self, x, H, W)
lib/VIT/encoder/pvtv2_encoder.py:45
Method
forward
(self, x, H, W)
lib/VIT/encoder/pvtv2_encoder.py:93
Method
forward
(self, x, H, W)
lib/VIT/encoder/pvtv2_encoder.py:150
Method
forward
(self, x)
lib/VIT/encoder/pvtv2_encoder.py:191
Method
forward
(self, x)
lib/VIT/encoder/pvtv2_encoder.py:360
Method
forward
(self, x, H, W)
lib/VIT/encoder/pvtv2_encoder.py:376
Method
forward
(self, x)
lib/VIT/encoder/swin_encoder.py:27
Method
forward
Args: x: input features with shape of (num_windows*B, N, C) mask: (0/-inf) mask with shape of (num_windows, Wh*Ww, Wh
lib/VIT/encoder/swin_encoder.py:116
Method
forward
(self, x)
lib/VIT/encoder/swin_encoder.py:236
Method
forward
x: B, H*W, C
lib/VIT/encoder/swin_encoder.py:310
Method
forward
(self, x)
lib/VIT/encoder/swin_encoder.py:391
Method
forward
(self, x)
lib/VIT/encoder/swin_encoder.py:443
Method
forward
(self, x)
lib/VIT/encoder/swin_encoder.py:590
Method
forward
(self, x)
lib/VIT/decoder/decoder_p.py:56
Method
forward
(self, x)
lib/VIT/decoder/decoder_p.py:73
Method
forward
(self, x)
lib/VIT/decoder/decoder_p.py:89
Method
forward
(self, x)
lib/VIT/decoder/decoder_p.py:104
Method
forward
(self, x,mask=None)
lib/VIT/decoder/decoder_p.py:130
← previous
next →
201–300 of 357, ranked by callers