Index

_

__call__() (AdversarialLoss method) (DiscriminatorLoss method) (MeanAbsoluteError method) (MeanSquaredError method) (RaAdversarialLoss method) (RaDiscriminatorLoss method) (VGGLoss method)

A

adjust_brightness() (in module simple_sr.utils.image.image_transforms) adjust_contrast() (in module simple_sr.utils.image.image_transforms) adjust_gamma() (in module simple_sr.utils.image.image_transforms) adjust_hue() (in module simple_sr.utils.image.image_transforms) adjust_jpg_quality() (in module simple_sr.utils.image.image_transforms) adjust_saturation() (in module simple_sr.utils.image.image_transforms)

AdversarialLoss (class in simple_sr.utils.models.loss_functions.adversarial_loss) after_epoch() (SRModel method) after_train_batch() (SRModel method) after_training() (SRModel method) after_validation_batch() (SRModel method) augment_img() (in module simple_sr.utils.image.image_transforms)

B

batch_history() (SRModel method) batch_metrics() (SRModel method) batch_summary_writer() (SRModel method) before_epoch() (SRModel method)

build_discriminator() (in module simple_sr.utils.models.model_builder) build_enhanced_resnet() (in module simple_sr.utils.models.model_builder) build_resnet() (in module simple_sr.utils.models.model_builder) build_vgg_16() (in module simple_sr.utils.models.model_builder) build_vgg_19() (in module simple_sr.utils.models.model_builder)

C

calculate_train_loss() (Discriminator method) (Generator method) calculate_validation_loss() (Discriminator method) (Generator method) combine_halfs() (in module simple_sr.utils.image.image_utils)

ConfigUtil (class in simple_sr.utils.config.config_util) critic_train_batch() (Discriminator method) critic_validation_batch() (Discriminator method) crop_divers() (in module simple_sr.utils.image.image_transforms) crop_naive() (in module simple_sr.utils.image.image_transforms)

D

DataPipeline (class in simple_sr.data_pipeline.data_pipeline) denormalize_11() (in module simple_sr.utils.image.image_transforms) Discriminator (class in simple_sr.models.discriminator)

discriminator() (SRModel method) discriminator_optimizer() (SRModel method) DiscriminatorLoss (class in simple_sr.utils.models.loss_functions.discriminator_loss)

E

epoch_history() (SRModel method) epoch_metrics() (SRModel method) epoch_summary_writer() (SRModel method) esrgan_generator() (Generator static method)

eval_pipeline() (DataPipeline static method) evaluate_on_testdata() (in module simple_sr.operations.evaluation) evaluate_on_validationdata() (in module simple_sr.operations.evaluation) evaluation_config() (ConfigUtil static method)

F

flip_along_x() (in module simple_sr.utils.image.image_transforms) flip_along_y() (in module simple_sr.utils.image.image_transforms) formatted_epoch_metrics() (Discriminator method) (Generator method) (SRModel method)

from_config() (DataPipeline static method) from_yaml() (ConfigUtil static method) (Discriminator static method) (Generator static method)

G

generate() (Generator method) Generator (class in simple_sr.models.generator)

generator() (SRModel method) generator_optimizer() (SRModel method) get_all_available_augmentations() (in module simple_sr.utils.image.image_transforms)

I

inference_pipeline() (DataPipeline static method) init() (SRModel static method)

initialize_relativistic() (Discriminator static method) initialize_standard() (Discriminator static method)

L

latest_checkpoint() (SRModel method)

loss_function() (Discriminator method) loss_functions() (Generator method)

M

MeanAbsoluteError (class in simple_sr.utils.models.loss_functions.mean_absolute_error) MeanSquaredError (class in simple_sr.utils.models.loss_functions.mean_squared_error) model() (Discriminator method) (Generator method) module simple_sr.operations.evaluation simple_sr.operations.training simple_sr.utils.config.config_util simple_sr.utils.image.image_transforms simple_sr.utils.image.image_utils simple_sr.utils.image.metrics simple_sr.utils.models.loss_functions.adversarial_loss simple_sr.utils.models.loss_functions.discriminator_loss simple_sr.utils.models.loss_functions.mean_absolute_error simple_sr.utils.models.loss_functions.mean_squared_error simple_sr.utils.models.loss_functions.ra_adversarial_loss simple_sr.utils.models.loss_functions.ra_discriminator_loss simple_sr.utils.models.loss_functions.vgg_loss simple_sr.utils.models.model_builder

N

normalize_01() (in module simple_sr.utils.image.image_transforms)

normalize_11() (in module simple_sr.utils.image.image_transforms)

P

prepare_image_grid() (in module simple_sr.utils.image.image_utils)

psnr() (in module simple_sr.utils.image.metrics) psnr_on_y() (in module simple_sr.utils.image.metrics)

R

RaAdversarialLoss (class in simple_sr.utils.models.loss_functions.ra_adversarial_loss) RaDiscriminatorLoss (class in simple_sr.utils.models.loss_functions.ra_discriminator_loss) read_img() (in module simple_sr.utils.image.image_utils) reconstruct_from_overlapping_patches() (in module simple_sr.utils.image.image_utils) reconstruct_from_patches() (in module simple_sr.utils.image.image_utils)

reset_batch_metrics() (Discriminator method) reset_epoch_metrics() (Discriminator method) resize() (in module simple_sr.utils.image.image_transforms) rotate90() (in module simple_sr.utils.image.image_transforms) rrdb() (Generator static method) run_training() (in module simple_sr.operations.training)

S

save_model() (SRModel method) save_single() (in module simple_sr.utils.image.image_utils) segment_into_patches() (in module simple_sr.utils.image.image_utils) simple_sr.operations.evaluation module simple_sr.operations.training module simple_sr.utils.config.config_util module simple_sr.utils.image.image_transforms module simple_sr.utils.image.image_utils module simple_sr.utils.image.metrics module simple_sr.utils.models.loss_functions.adversarial_loss module simple_sr.utils.models.loss_functions.discriminator_loss module

simple_sr.utils.models.loss_functions.mean_absolute_error module simple_sr.utils.models.loss_functions.mean_squared_error module simple_sr.utils.models.loss_functions.ra_adversarial_loss module simple_sr.utils.models.loss_functions.ra_discriminator_loss module simple_sr.utils.models.loss_functions.vgg_loss module simple_sr.utils.models.model_builder module srgan_generator() (Generator static method) SRModel (class in simple_sr.models.sr_model) srresnet() (Generator static method) ssim() (in module simple_sr.utils.image.metrics) stop_early() (SRModel method)

T

tensor_to_img() (in module simple_sr.utils.image.image_utils) test_and_plot() (SRModel method) test_batch_generator() (DataPipeline method)

train_batch_generator() (DataPipeline method) train_step() (SRModel method) training_config() (ConfigUtil static method)

V

validation_batch_generator() (DataPipeline method)

validation_step() (SRModel method) VGGLoss (class in simple_sr.utils.models.loss_functions.vgg_loss)