Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JizhiziLi/GFM
/ functions
Functions
66 in github.com/JizhiziLi/GFM
⨍
Functions
66
◇
Types & classes
6
↓ 60 callers
Function
build_decoder
(in_channels, mid_channels_1, mid_channels_2, out_channels, last_bnrelu, upsample_flag)
core/gfm.py:44
↓ 18 callers
Function
conv_up_psp
(in_channels, out_channels, up_sample)
core/gfm.py:25
↓ 11 callers
Function
extract_pure_name
(original_name)
core/util.py:23
↓ 6 callers
Function
get_alpha_loss
(predict, alpha, trimap)
core/evaluate.py:37
↓ 6 callers
Function
get_crossentropy_loss
(output_class, gt,pre)
core/evaluate.py:22
↓ 6 callers
Function
get_laplacian_loss
(predict, alpha, trimap)
core/evaluate.py:87
↓ 6 callers
Function
listdir_nohidden
(path)
core/util.py:27
↓ 5 callers
Function
get_alpha_loss_whole_img
(predict, alpha)
core/evaluate.py:48
↓ 5 callers
Function
get_composition_loss_whole_img
(img, alpha, fg, bg, predict)
core/evaluate.py:109
↓ 5 callers
Function
get_laplacian_loss_whole_img
(predict, alpha)
core/evaluate.py:100
↓ 4 callers
Function
build_bb
(in_channels, mid_channels, out_channels)
core/gfm.py:32
↓ 4 callers
Function
check_if_folder_exists
(folder_path)
core/util.py:39
↓ 4 callers
Function
collaborative_matting
(rosta, glance_sigmoid, focus_sigmoid)
core/util.py:62
↓ 4 callers
Function
inference_img_scale
(args, model, scale_img)
core/test.py:51
↓ 4 callers
Function
laplacian_pyramid
(img, kernel, max_levels=5)
core/evaluate.py:76
↓ 4 callers
Function
process_fgbg
(ori, mask, is_fg, fgbg_path=None)
core/util.py:182
↓ 3 callers
Method
__init__
(self, args)
core/gfm.py:129
↓ 3 callers
Function
refresh_folder
(folder_path)
core/util.py:43
↓ 3 callers
Function
resize_img
(ori, img)
core/util.py:178
↓ 2 callers
Function
build_gauss_kernel
(size=5, sigma=1.0, n_channels=1, cuda=False)
core/evaluate.py:59
↓ 2 callers
Function
conv3x3
3x3 convolution with padding
core/gfm.py:20
↓ 2 callers
Function
inference_img_gfm
(args, model, img, option)
core/test.py:79
↓ 2 callers
Function
save_test_result
(save_dir, predict)
core/util.py:115
↓ 1 callers
Method
__init__
(self, args, transform)
core/data.py:68
↓ 1 callers
Method
_make_stage
(self, features, size)
core/gfm.py:99
↓ 1 callers
Function
add_guassian_noise
(img, fg, bg)
core/util.py:190
↓ 1 callers
Function
calculate_sad_fgbg
(predict, alpha, trimap)
core/evaluate.py:140
↓ 1 callers
Function
calculate_sad_mse_mad
(predict_old,alpha,trimap)
core/evaluate.py:121
↓ 1 callers
Function
calculate_sad_mse_mad_whole_img
(predict, alpha)
core/evaluate.py:133
↓ 1 callers
Function
compute_connectivity_loss_whole_image
(pd, gt, step=0.1)
core/evaluate.py:167
↓ 1 callers
Function
compute_gradient_whole_image
(pd, gt)
core/evaluate.py:153
↓ 1 callers
Function
conv_gauss
convolve img with a gaussian kernel that has been built with build_gauss_kernel
core/evaluate.py:70
↓ 1 callers
Function
create_folder_if_not_exists
(folder_path)
core/util.py:35
↓ 1 callers
Function
format_second
(secs)
core/train.py:72
↓ 1 callers
Function
gen_bw_from_segmap_e2e
(segmap)
core/util.py:109
↓ 1 callers
Function
gen_dilate
(alpha, kernel_size)
core/util.py:241
↓ 1 callers
Function
gen_erosion
(alpha, kernel_size)
core/util.py:247
↓ 1 callers
Function
gen_trimap_from_segmap_e2e
(segmap)
core/util.py:102
↓ 1 callers
Function
gen_trimap_with_dilate
(alpha, kernel_size)
core/util.py:232
↓ 1 callers
Function
generate_composite_coco
(fg, bg, mask)
core/util.py:222
↓ 1 callers
Function
generate_composite_img
(img, alpha_channel)
core/util.py:50
↓ 1 callers
Function
generate_composite_rssn
(fg, bg, mask, fg_denoise=None, bg_denoise=None)
core/util.py:201
↓ 1 callers
Function
generate_paths_for_dataset
(args)
core/util.py:123
↓ 1 callers
Function
get_args
()
core/train.py:29
↓ 1 callers
Function
get_args
()
core/test.py:26
↓ 1 callers
Function
get_masked_local_from_global_test
(global_result, local_result)
core/util.py:95
↓ 1 callers
Function
load_dataset
(args)
core/train.py:56
↓ 1 callers
Function
load_model
(args)
core/train.py:62
↓ 1 callers
Function
load_model_and_deploy
(args)
core/test.py:275
↓ 1 callers
Function
main
()
core/train.py:159
↓ 1 callers
Function
save_last_checkpoint
(args, model)
core/train.py:152
↓ 1 callers
Function
test_am2k
(args, model)
core/test.py:138
↓ 1 callers
Function
test_samples
(args, model)
core/test.py:229
↓ 1 callers
Function
train
(args, model, optimizer, train_loader, epoch)
core/train.py:79
↓ 1 callers
Function
trim_img
(img)
core/util.py:173
Method
__call__
(self, *argv)
core/data.py:36
Method
__getitem__
(self,index)
core/data.py:83
Method
__init__
(self, inplanes, planes, stride=1, downsample=None)
core/gfm.py:67
Method
__init__
(self, features, out_features=1024, sizes=(1, 2, 3, 6))
core/gfm.py:92
Method
__init__
(self, channel, reduction=4)
core/gfm.py:111
Method
__init__
(self)
core/data.py:33
Method
__len__
(self)
core/data.py:125
Method
forward
(self, x)
core/gfm.py:77
Method
forward
(self, feats)
core/gfm.py:104
Method
forward
(self, x)
core/gfm.py:121
Method
forward
(self, input)
core/gfm.py:343