Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ShawnBIT/UNet-family
/ functions
Functions
12 in github.com/ShawnBIT/UNet-family
⨍
Functions
12
◇
Types & classes
4
↓ 6 callers
Function
init_weights
(net, init_type='normal')
utils.py:5
↓ 2 callers
Function
count_param
(model)
utils.py:24
↓ 1 callers
Method
__init__
(self, in_size, out_size, is_deconv, n_concat=2)
layers.py:42
↓ 1 callers
Function
add_path
(path)
networks/_init_paths.py:5
Method
__init__
(self, in_size, out_size, is_batchnorm, n=2, ks=3, stride=1, padding=1)
layers.py:6
Method
__init__
(self, in_channels=1, n_classes=2, feature_scale=2, is_deconv=True, is_batchnorm=True, is_ds=True)
networks/UNet_Nested.py:9
Method
__init__
(self, in_channels=1, n_classes=2, feature_scale=2, is_deconv=True, is_batchnorm=True)
networks/UNet.py:9
Method
forward
(self, inputs)
layers.py:33
Method
forward
(self, high_feature, *low_feature)
layers.py:57
Method
forward
(self, inputs)
networks/UNet_Nested.py:56
Method
forward
(self, inputs)
networks/UNet.py:42
Function
weights_init_kaiming
(m)
utils.py:12