MCPcopy Create free account

hub / github.com/1343744768/Multiattention-UNet / functions

Functions65 in github.com/1343744768/Multiattention-UNet

↓ 13 callersMethodrand
(self, a=0, b=1)
utils/dataloader.py:95
↓ 7 callersFunctionCE_Loss
(inputs, target, cls_weights, num_classes=21)
nets/unet_training.py:6
↓ 7 callersFunctionDice_loss
(inputs, target, beta=1, smooth = 1e-5)
nets/unet_training.py:35
↓ 7 callersFunctionFocal_Loss
(inputs, target, cls_weights, num_classes=21, alpha=0.5, gamma=2)
nets/unet_training.py:18
↓ 7 callersFunctionf_score
(inputs, target, beta=1, smooth = 1e-5, threhold = 0.5)
utils/utils_metrics.py:12
↓ 6 callersMethod__init__
(self, num_classes=21)
nets/unet.py:183
↓ 6 callersFunctionget_lr
(optimizer)
utils/utils.py:26
↓ 4 callersMethodappend_loss
(self, loss, val_loss = 0)
utils/callback.py:20
↓ 4 callersFunctioncvtColor
(image)
utils/utils.py:4
↓ 4 callersFunctiondraw_plot_func
(values, name_classes, plot_title, x_label, output_path, tick_font_size = 12, plt_show = True)
utils/utils_metrics.py:134
↓ 4 callersFunctionpreprocess_input
(image)
utils/utils.py:30
↓ 3 callersFunctioncompute_mIoU
(gt_dir, pred_dir, png_name_list, num_classes, name_classes)
utils/utils_metrics.py:57
↓ 3 callersMethodget_miou_png
(self, image)
unet.py:107
↓ 3 callersFunctionresize_image
(image, size)
utils/utils.py:11
↓ 3 callersFunctionshow_results
(miou_out_path, hist, IoUs, PA_Recall, Precision, name_classes, tick_font_size = 12)
utils/utils_metrics.py:156
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
nets/unet.py:113
↓ 2 callersFunctionper_Accuracy
(hist)
utils/utils_metrics.py:54
↓ 2 callersFunctionper_class_PA_Recall
(hist)
utils/utils_metrics.py:48
↓ 2 callersFunctionper_class_iu
(hist)
utils/utils_metrics.py:45
↓ 2 callersFunctionweights_init
(net, init_type='normal', init_gain=0.02)
nets/unet_training.py:55
↓ 1 callersMethod__init__
(self, limit=90, prob=0.7)
utils/dataloader.py:10
↓ 1 callersFunctionadjust_axes
(r, t, fig, axes)
utils/utils_metrics.py:125
↓ 1 callersMethoddetect_image
(self, image)
unet.py:51
↓ 1 callersFunctionfast_hist
(a, b, n)
utils/utils_metrics.py:34
↓ 1 callersFunctionfit_one_epoch
(model_train, model, loss_history, optimizer, epoch, epoch_step, epoch_step_val, gen, gen_va
utils/utils_fit_multi.py:9
↓ 1 callersFunctionfit_one_epoch
(model_train, model, loss_history, optimizer, epoch, epoch_step, epoch_step_val, gen, gen_val, Epoch, cuda, di
utils/utils_fit.py:9
↓ 1 callersMethodgenerate
(self)
unet.py:39
↓ 1 callersMethodget_lr
we will use the first m batches, and set the learning rate to base_lr * m / total_iters
train_multi_GPU.py:33
↓ 1 callersMethodget_random_data
(self, image, label, input_shape, jitter=.3, hue=.1, sat=1.5, val=1.5, random=True)
utils/dataloader.py:98
↓ 1 callersMethodloss_plot
(self)
utils/callback.py:33
↓ 1 callersMethodloss_plot
(self)
utils/callbacks.py:34
↓ 1 callersFunctionmain
(args)
train_multi_GPU.py:40
↓ 1 callersFunctionper_class_Precision
(hist)
utils/utils_metrics.py:51
Method__call__
(self, img, mask=None)
utils/dataloader.py:14
Method__call__
(self, img, mask=None)
utils/dataloader.py:41
Method__getitem__
(self, index)
utils/dataloader.py:78
Method__init__
(self, **kwargs)
unet.py:24
Method__init__
(self, optimizer, total_iters, last_epoch=-1)
train_multi_GPU.py:29
Method__init__
(self, log_dir, val_loss_flag = True)
utils/callback.py:8
Method__init__
(self, max_h_size=20, max_w_size=20, fill_value=255, prob=0.5)
utils/dataloader.py:34
Method__init__
(self, annotation_lines, input_shape, num_classes, train, dataset_path)
utils/dataloader.py:66
Method__init__
(self, log_dir, args, val_loss_flag = True)
utils/callbacks.py:8
Method__init__
(self, dim, # 输入token的dim num_heads=8, qkv_bias=False,
nets/unet.py:7
Method__init__
(self, channels=None, e_lambda=1e-4)
nets/unet.py:50
Method__init__
(self, in_planes, ratio=16)
nets/unet.py:77
Method__init__
(self, kernel_size=7)
nets/unet.py:96
Method__init__
(self, inplanes, planes)
nets/unet.py:118
Method__init__
(self, in_size, mid_size, out_size, upsize, att_type='channel')
nets/unet.py:147
Method__len__
(self)
utils/dataloader.py:75
Method__repr__
(self)
nets/unet.py:56
Method_initialize_weights
(self, *stages)
nets/unet.py:254
Methodappend_loss
(self, loss, val_loss = 0)
utils/callbacks.py:21
Functionfit_one_epoch_no_val
(model_train, model, loss_history, optimizer, epoch, epoch_step, gen, Epoch, cuda, dice_loss,
utils/utils_fit_multi.py:175
Functionfit_one_epoch_no_val
(model_train, model, loss_history, optimizer, epoch, epoch_step, gen, Epoch, cuda, dice_loss, focal_loss, cls_
utils/utils_fit.py:108
Methodforward
(self, x)
nets/unet.py:23
Methodforward
(self, x)
nets/unet.py:65
Methodforward
(self, x)
nets/unet.py:88
Methodforward
(self, x)
nets/unet.py:105
Methodforward
(self, x)
nets/unet.py:130
Methodforward
(self, inputs1, inputs2)
nets/unet.py:174
Methodforward
(self, inputs)
nets/unet.py:233
Methodget_FPS
(self, image, test_interval)
unet.py:81
Methodget_module_name
()
nets/unet.py:62
Functioninit_func
(m)
nets/unet_training.py:56
Functionunet_dataset_collate
(batch)
utils/dataloader.py:170