To develop a convolutional autoencoder for image denoising application.
Include the neural network model diagram.
Write your own steps
# Autoencoder Definition
class DenoisingAutoencoder(nn.Module):
def __init__(self):
# Initialize model
# Training function
# Visualization function
Include your model summary
Include a few sample images here.
Include your result here