MCPcopy Create free account

hub / github.com/RylonW/DocNLC / types & classes

Types & classes48 in github.com/RylonW/DocNLC

↓ 35 callersClassSSIMLoss
models/loss_new.py:54
↓ 9 callersClassCharbonnierLoss
Charbonnier Loss (L1)
models/loss.py:5
↓ 5 callersClassConvLSTM
models/multitask_BCDU_model.py:71
↓ 3 callersClassConvBLSTM
models/multitask_BCDU_model.py:107
↓ 2 callersClassAttING
models/archs/EnhanceN_arch.py:22
↓ 2 callersClassCLSTM_cell
Initialize a basic Conv LSTM cell. Args: shape: int tuple thats the height and width of the hidden states h and c() filter_size: int t
models/archs/layer.py:18
↓ 2 callersClassDEGAN
models/multitask_DeGAN_model.py:26
↓ 2 callersClassKernel
[summary] Class representing a motion blur kernel of a given intensity. [description] Keyword Arguments: size {tuple} -- Size
utils/motionblur.py:52
↓ 1 callersClassAttING
models/multitask_docnc_model.py:40
↓ 1 callersClassAttING
models/multitask_Barlow_model_new.py:41
↓ 1 callersClassBCDUNet
models/multitask_BCDU_model.py:126
↓ 1 callersClassBarlowTwins
models/multitask_Barlow_model_new.py:190
↓ 1 callersClassBarlow_Loss
models/multitask_Barlow_model_new.py:240
↓ 1 callersClassBinModel
The autoencoder model to enhance images in an image to image translation fashion. This code is built on top of the vit-pytorch code https://g
models/multitask_ViT_model.py:30
↓ 1 callersClassConvLSTMCell
models/multitask_BCDU_model.py:27
↓ 1 callersClassCosineAnnealingLR_Restart
models/lr_scheduler.py:35
↓ 1 callersClassDIAE
models/multitask_DIAE_model.py:40
↓ 1 callersClassDocNC
models/multitask_docnc_model.py:82
↓ 1 callersClassMultiStepLR_Restart
models/lr_scheduler.py:8
↓ 1 callersClassNoneDict
options/options.py:84
↓ 1 callersClassSeeInDark
models/multitask_Barlow_model_new.py:83
↓ 1 callersClassUNet
models/multitask_UNet_model.py:26
↓ 1 callersClassVgg19
models/loss_new.py:95
ClassBaseModel
models/base_model.py:8
ClassBasicBlock
pytorch torch.nn.Linear wrapper Notes --------------------------------------------------------------------- use partial() to wrap activati
models/archs/layer.py:244
ClassCLSTM
Initialize a basic Conv LSTM cell. Args: filter_size: int that is the height and width of the filters num_features: int thats the num
models/archs/layer.py:61
ClassDatasetFromFolder
For test dataset, specify `group_file` parameter to target TXT file data_augmentation = None black_edge_crop = None flip = None
data/SIEN_dataset.py:133
ClassDatasetFromFolder
For test dataset, specify `group_file` parameter to target TXT file data_augmentation = None black_edge_crop = None flip = None
data/multitask_dataset.py:142
ClassDiscriminator_VGG_128
models/archs/discriminator_vgg_arch.py:6
ClassDownsample
models/archs/arch_util.py:19
ClassDownsample1D
models/archs/arch_util.py:73
ClassGANLoss
models/loss.py:19
ClassGradientLoss
Gradient Histogram Loss
models/loss_new.py:153
ClassGradientPenaltyLoss
models/loss.py:54
ClassInvertibleConv1x1
models/archs/arch_util.py:129
ClassProgressBar
A progress bar which can print the progress modified from https://github.com/hellock/cvbase/blob/master/cvbase/progress.py
utils/util.py:291
ClassSIEN_Model
models/SIEN_model.py:26
ClassSeeInDark
models/archs/EnhanceN_arch.py:64
ClassVGGFeatureExtractor
models/archs/discriminator_vgg_arch.py:62
ClassVGGLoss
models/loss_new.py:132
Classmultitask_BCDUNet_model
models/multitask_BCDU_model.py:199
Classmultitask_Barlow_model
models/multitask_Barlow_model_new.py:296
Classmultitask_DEGAN_model
models/multitask_DeGAN_model.py:133
Classmultitask_DIAE_model
models/multitask_DIAE_model.py:255
Classmultitask_SIEN_model
models/multitask_SIEN_model.py:27
Classmultitask_UNet_model
models/multitask_UNet_model.py:133
Classmultitask_ViT_model
models/multitask_ViT_model.py:136
Classmultitask_docnc_model
models/multitask_docnc_model.py:194