Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Andy-zhujunwen/UNET-ZOO
/ functions
Functions
143 in github.com/Andy-zhujunwen/UNET-ZOO
⨍
Functions
143
◇
Types & classes
50
↓ 17 callers
Method
__init__
(self, in_ch, out_ch, bilinear=True)
cenet.py:504
↓ 8 callers
Function
_vgg
(arch, cfg, batch_norm, pretrained, progress, **kwargs)
fcn.py:93
↓ 5 callers
Method
__init__
(self, features, num_classes=1000, init_weights=True)
fcn.py:32
↓ 3 callers
Method
__init__
(self, in_ch, out_ch)
UNet.py:25
↓ 3 callers
Method
__init__
(self,ch_in,ch_out)
r2unet.py:6
↓ 3 callers
Method
__init__
(self,ch_in,ch_out)
attention_unet.py:22
↓ 2 callers
Method
__init__
(self, channels_high, channels_low, upsample=True)
channel_unet.py:6
↓ 2 callers
Function
get_dice
(mask_name,predict)
metrics.py:84
↓ 2 callers
Function
get_hd
(mask_name,predict)
metrics.py:116
↓ 2 callers
Function
get_iou
(mask_name,predict)
metrics.py:38
↓ 2 callers
Function
make_layers
(cfg, batch_norm=False)
fcn.py:69
↓ 2 callers
Function
metrics_plot
(arg,name,*args)
plot.py:15
↓ 1 callers
Method
__init__
(self, in_ch, out_ch)
unetpp.py:7
↓ 1 callers
Method
_fast_hist
(self, label_pred, label_true)
metrics.py:16
↓ 1 callers
Method
_initialize_weights
(self)
fcn.py:55
↓ 1 callers
Method
_make_stage
(self, features, size)
cenet.py:107
↓ 1 callers
Function
getArgs
()
main.py:24
↓ 1 callers
Method
getDataPath
(self)
dataset.py:24
↓ 1 callers
Method
getDataPath
(self)
dataset.py:71
↓ 1 callers
Method
getDataPath
(self)
dataset.py:120
↓ 1 callers
Method
getDataPath
(self)
dataset.py:171
↓ 1 callers
Method
getDataPath
(self)
dataset.py:226
↓ 1 callers
Method
getDataPath
(self)
dataset.py:286
↓ 1 callers
Method
getDataPath
(self)
dataset.py:342
↓ 1 callers
Function
getDataset
(args)
main.py:78
↓ 1 callers
Function
getLog
(args)
main.py:40
↓ 1 callers
Function
getModel
(args)
main.py:52
↓ 1 callers
Function
get_fcn8s
(n_class=1)
fcn.py:412
↓ 1 callers
Function
loss_plot
(args,loss)
plot.py:3
↓ 1 callers
Function
test
(val_dataloaders,save_predict=False)
main.py:219
↓ 1 callers
Function
train
(model, criterion, optimizer, train_dataloader,val_dataloader, args)
main.py:163
↓ 1 callers
Function
val
(model,best_iou,val_dataloaders)
main.py:128
Method
__getitem__
(self, index)
dataset.py:44
Method
__getitem__
(self, index)
dataset.py:90
Method
__getitem__
(self, index)
dataset.py:133
Method
__getitem__
(self, index)
dataset.py:188
Method
__getitem__
(self, index)
dataset.py:241
Method
__getitem__
(self, index)
dataset.py:304
Method
__getitem__
(self, index)
dataset.py:356
Method
__init__
(self,input_nbr,label_nbr)
segnet.py:7
Method
__init__
(self, args,in_channel,out_channel)
unetpp.py:41
Method
__init__
(self, in_ch, out_ch)
channel_unet.py:47
Method
__init__
(self, in_ch, out_ch)
channel_unet.py:63
Method
__init__
(self, pretrained_net, n_class)
fcn.py:194
Method
__init__
(self, pretrained_net, n_class)
fcn.py:237
Method
__init__
(self, pretrained_net, n_class)
fcn.py:272
Method
__init__
(self, pretrained_net, n_class)
fcn.py:309
Method
__init__
(self, pretrained=True, model='vgg16', requires_grad=True, remove_fc=True, show_params=False)
fcn.py:349
Method
__init__
(self, state, transform=None, target_transform=None)
dataset.py:15
Method
__init__
(self, state, transform=None, target_transform=None)
dataset.py:62
Method
__init__
(self, state, transform=None, target_transform=None)
dataset.py:108
Method
__init__
(self, state, transform=None, target_transform=None)
dataset.py:159
Method
__init__
(self, state, transform=None, target_transform=None)
dataset.py:214
Method
__init__
(self, state, transform=None, target_transform=None)
dataset.py:274
Method
__init__
(self, state, transform=None, target_transform=None)
dataset.py:330
Method
__init__
(self, num_classes)
metrics.py:12
Method
__init__
(self, channel)
cenet.py:14
Method
__init__
(self, channel)
cenet.py:35
Method
__init__
(self, channel)
cenet.py:56
Method
__init__
(self, channel)
cenet.py:77
Method
__init__
(self, features, out_features=1024, sizes=(2, 3, 6, 14))
cenet.py:100
Method
__init__
(self, in_channels)
cenet.py:120
Method
__init__
(self, in_channels, n_filters)
cenet.py:142
Method
__init__
(self, num_classes=1, num_channels=3)
cenet.py:171
Method
__init__
(self, num_classes=1, num_channels=3)
cenet.py:229
Method
__init__
(self, num_classes=1, num_channels=3)
cenet.py:287
Method
__init__
(self, num_classes=1, num_channels=3)
cenet.py:345
Method
__init__
(self, num_classes=12, num_channels=3)
cenet.py:404
Method
__init__
(self, in_ch, out_ch)
cenet.py:464
Method
__init__
(self, in_ch, out_ch)
cenet.py:481
Method
__init__
(self, in_ch, out_ch)
cenet.py:491
Method
__init__
(self, in_ch, out_ch)
cenet.py:524
Method
__init__
(self, n_channels=3, n_classes=1)
cenet.py:534
Method
__init__
(self, in_ch, out_ch)
UNet.py:9
Method
__init__
(self, in_channels, n_filters)
UNet.py:81
Method
__init__
(self, num_classes=1, num_channels=3,pretrained=True)
UNet.py:109
Method
__init__
(self, ch_out, t=2)
r2unet.py:19
Method
__init__
(self,ch_in,ch_out,t=2)
r2unet.py:39
Method
__init__
(self, img_ch=3, output_ch=1, t=2)
r2unet.py:52
Method
__init__
(self,ch_in,ch_out)
attention_unet.py:7
Method
__init__
(self, F_g, F_l, F_int)
attention_unet.py:36
Method
__init__
(self, img_ch=3, output_ch=1)
attention_unet.py:70
Method
__len__
(self)
dataset.py:58
Method
__len__
(self)
dataset.py:104
Method
__len__
(self)
dataset.py:155
Method
__len__
(self)
dataset.py:210
Method
__len__
(self)
dataset.py:270
Method
__len__
(self)
dataset.py:326
Method
__len__
(self)
dataset.py:378
Method
add_batch
(self, predictions, gts)
metrics.py:23
Method
evaluate
(self)
metrics.py:27
Method
forward
(self, x)
segnet.py:74
Method
forward
(self, input)
unetpp.py:18
Method
forward
(self, input)
unetpp.py:80
Method
forward
(self, fms_high, fms_low, fm_mask=None)
channel_unet.py:24
Method
forward
(self, input)
channel_unet.py:58
Method
forward
(self, x)
channel_unet.py:89
Method
forward
(self, x)
fcn.py:48
Method
forward
(self, x)
fcn.py:211
Method
forward
(self, x)
fcn.py:254
next →
1–100 of 143, ranked by callers