def mae_albdiv_bn(train_data, test_data):
"""Train the MAE denoiser with albedo divide and batch norm on. Use all
features."""
config.ALBEDO_DIVIDE = True
feature_list = ["normal", "albedo", "depth"]
denoiser = Denoiser(
train_data,
test_data,
adam_lr=1e-4,
num_epochs=
100000, # Stupid number of epochs, early stopping will prevent reaching this
early_stopping=True,
kernel_predict=True,
bn=True,
feature_list=feature_list,
num_layers=7,
batch_size=64,
loss="mae",
model_dir="../experiments/models/mae_albdiv_bn",
log_dir="../experiments/logs/mae_albdiv_bn")
denoiser.buildNetwork()
denoiser.train()
def vgg_albdiv_bn(train_data, test_data):
"""Train the vgg denoiser with albedo divide and batch norm. Use all
features"""
config.ALBEDO_DIVIDE = True
feature_list = ["normal", "albedo", "depth"]
denoiser = Denoiser(
train_data,
test_data,
adam_lr=1e-3,
num_epochs=100000,
early_stopping=True,
kernel_predict=True,
bn=True,
feature_list=feature_list,
num_layers=7,
batch_size=64,
loss="vgg22",
model_dir="../experiments/models/vgg_albdiv_bn",
log_dir="../experiments/logs/vgg_albdiv_bn",
)
denoiser.buildNetwork()
denoiser.train()
