Photorealistic rendering of painting + Venice Underwater

Abstract

The main goal of our project is to automatically transform old paintings and drawings of Venice into a photorealistic representation of the past, thus create photos never taken. In order to achieve this goal, we will utilize style transfer generative adversarial networks to learn what separates a drawing or painting from a photograph and then utilize that model.

After a review of the current literature in the field, we settled on implementing the relatively new contrastive unpaired translation GAN with some key modifications. In particular we plan to build a new auxiliary loss function to enforce geometric continuity between generated and original building, as well as add spectral norm and self-attention layers to allow use to effectually expand the size of the network. The target data will be a dataset of real photos of Venice, with datasets of either b/w drawings, or b/w paintings, or coloured paintings as the input. Finally, the trained models should take a drawing or painting as input and deliver its photorealistic representation as output.

As a subgoal of the project, another GAN will be trained to create a visual representation of Venice as an underwater city. This representation takes the rising sea level into account and allows to scale time into the future by visualizing the drowning of the world heritage.

Milestones

Milestone 1

• Choose GAN architecture to use as base architecture for the project
  • Selected architecture: Contrastive Unpaired Translation (CUT)
• Data collection (web scraping and CINI foundation):
  • Color paintings
  • Monochrome paintings
  • Monochrome drawings
  • Underwater images
  • Modern photos of Venice

Milestone 2

• Clean collected images:
  • Restrict images to those that show more "standard" subjects: outdoor images of Venice
  • Crop out image frames, watermarks
• Test initial results of CUT model without hyperparameter tuning

Milestone 3

• Perform hyperparameter tuning on CUT model for each of the final models (color, monochrome, underwater)
• Look into model extensions, in order to enhance learning ability
  • Additions: Canny edge detection, spectral loss, self-attention layers

Milestone 4

• Find optimal parameters for each model
• Implement front end in order to visualize the model predicted results
  • Same front end for color and monochrome models: displays input image and output image
  • Underwater model front end: displays input image and output image with a slider representing the rising sea level

Planning

Week	Tasks
9.11.2020 - 15.11.2020 (week 9)	Midterm presentation ✔️ Test edge loss ✔️ Prepare sweep visualization ✔️
16.11.2020 - 22.11.2020 (week 10)	Test edge loss (cont) ✔️ Prepare project Wiki page ✔️ Perform sweep (1) (params: lambda_GAN = [0.8, 1.0, 1.2], lambda_NCE = [0.8, 1.0, 1.2]) ✔️ Perform sweep (2) (params: edgeLoss = [0.8, 1.6], canny1 = [150, 250], canny2 = [250, 400]) ✔️ Implement spectral loss ✔️ Scrape more color paintings ✔️ Scrape more underwater images ✔️ Clean colour photo data ✔️
23.11.2020 - 29.11.2020 (week 11)	Perform sweep (3) (params: n_epochs = [100, 250], n_epochs_decay = [100, 250]) ❗ (sweeps do not show model performance properly; replace with actual model testing) Addition ❗: Test model parameters: (n_epochs, n_epochs_decay): (100,300), (300,100), (200,400) ✔️ Test spectral loss ✔️ Test best performing models from sweeps ❗ (replacement: experiment further with epochs) Implement self-attention model ✔️ Clean underwater picture data ✔️ Test underwater model ❌
30.11.2020 - 6.12.2020 (week 12)	Test underwater model ✔️ Finish tests, select best hyperparameters/additions for each model Monochrome to photo ✔️ Colour to photo ✔️ Photo to underwater ❌ Clean up code ✔️ Implement front end ❌ Write report ✔️
7.12.2020 - 13.12.2020 (week 13)	Select best hyperparameters for underwater model ✔️ Clean up code (finish) Implement front end Finish writing report Prepare presentation
14.12.2020 - 16.12.2020 (week 14)	Prepare presentation Final project presentation ✨

Deliverables

TODO motivation, description of deliverables.

We invite the reader to see our project GitHub page [1] to test the results of our pretrained models with new data using our front end, or to train new models.

For the color and monochrome models, the front end implementation looks as in Figure TODO, showing the initial painting/drawing and the final photorealistic representation.

For the underwater Venice model, the front end implementation looks as in Figure TODO, showing the initial photo, and the underwater representation with a slider, to show how "far" underwater the city is.

The final training data used for each of the three models produced is located in the following folder: /dhlabdata4/makhmuto/DoGANa_datasets/.

Methodology

Data collection

One of the primary challenges of this project is the acquisition and processing of a sufficiently large, high quality and diverse dataset. Our dataset consists of color paintings of Venice, monochrome paintings and drawings of Venice, photos of Venice, and underwater photos. The two main data sources used are the following:

• CINI dataset (provided by EPFL DH lab)
  • ~800 monochrome drawings and graphics of Venice
  • 331 paintings of Venice
• Web scraping:
  • Sources: Google Images, Flickr
  • Data acquired:
    • ~1300 color paintings of Venice
    • ~1600 photos of Venice
    • ~700 images of underwater shipwrecks or sunken cities

Web scraping

We primarily use Selenium web driver for crawling, and urllib to download the source images. We used search terms such as “color painting of Venice” in English, Italian, French, Russian, and Mandarin. To make the model learn more Venetian properties of the paintings and photos, we overweighed the number of photos of landmarks in Venice, as the majority of paintings and drawings were of the same few landmarks, for instance, St Mark’s Square. Thus, we used additional search terms, such as “color painting st marks square” or “photo st marks square”.

Data processing

The data cleaning and processing component of the project was one of the most crucial steps, as the content and quality of our images defined what the GAN model was able to learn and reproduce. The main tasks of this part of the project were to parse the collected data for content, and to cut out irrelevant components.

Content parsing

We removed web scraped images that have less conventional portrayals of the city of Venice, or images that were in no way semantically related to the search terms. For instance, as we can see in Fig. 1, none of these images are standard depictions of a Venetian setting: amongst the desired images, we also obtain images of paintings from Venice (but not of Venice), paintings of Venetian settings that do not portray a larger image of the city (just windows or graffiti, for example). We thus removed such images, as the goal is to create a model that imitates a photorealistic Venice.

Fig. 1: Web scraped images relating to Venice, but that do not portray a standard Venice setting

Image processing

Some web scraped images had appropriate content but had frames or watermarks that we wanted to remove, as in Fig. 2. To do this, we cropped the images either manually or using the Python Image Library (PIL).

Fig. 2: Original painting with frame (left), and processed version with frame removed (right)

Final data

After the data cleaning and processing steps, we had the final dataset consisting of:

• 807 monochrome paintings and graphics of Venice
• 901 color paintings of Venice
• 605 photos of Venice
• 431 underwater images

In a further work, there would be potential to augment each of the datasets, but we were more focused on using architectural improvements to our GAN model to achieve better results.

Model selection

At the start of this project, we had planned to implement our model using the CycleGAN architecture, as it is a GAN architecture that allows to translate an image from a source domain X to a target domain Y without necessarily having paired samples in domains X and Y. While doing some initial research for the project, however, we found a different model that better suited our needs: the contrastive unpaired translation (CUT) model. The CUT model was presented at ECCV 2020 in August, and is based on the implementation of CycleGAN. This framework also allows for unpaired image-to-image translation, in addition to having a better performance than CycleGAN and a shorter training time. More details about the model architecture can be found in the paper [2].

Model optimization

In order to optimize the model, in the first stage we optimized some of the model hyperparameters, and then added some additional features, such as Canny edge detection, spectral loss and self-attention networks.

Original hyperparameter optimization

We tested the following hyperparameters of the original CUT model [3]

• Learning rate
• Number of epochs with initial learning rate
• Number of epochs with linearly decaying learning rate to zero

We observed that our model tended to overfit with higher learning rates, or with a high number of epochs with the initial learning rate. We thus made sure to do the following in subsequent experiments:

• Set learning rate to lower values (typically between 0.0001 and 0.00018); original default: 0.0002
• Set the number of epochs with the initial learning rate to be significantly lower than the number of epochs with a linearly decaying learning rate (typically 150:300); original default: 200:200

Edge detection

In the initial model results, we observed that some shapes were not being detected properly in both drawings, paintings and photos. This resulted in a worse performing model: some structures in paintings were removed, as their shapes were undetected, as we can see in Fig. 3. To combat this issue, we implemented an edge detection component into the model.

We use the Canny edge detection algorithm [4]. This algorithm allows us to extract structural information from the images, by applying a Gaussian filter and setting thresholds to filter the most representative images. Then, the structural similarity of the input and output images is compared, based on the edges extracted by the filter. A higher structural similarity allows for a better mapping between the images. We added this “edge loss” as a summand of the total model generator loss function, forcing the model to take into account the structural similarity between the input and output images in the learning process.

Fig. 3: Bad performance of edge detection in the initial model; some details are ignored by the model.

Spectral loss

Todo

Self-attention layers

Todo

Estimated Relative Sea-Level Rise in Venice

In order to reliably depict the rising sea level in the coming decades, we have consulted relevant research literature.

To predict the sea-level rise in Venice, several variables have to be taken into account. The natural and anthropogenic land subsidence of Venice*4, the rising sea level due to climate change and the additional sinking due to increased groundwater extraction (excessively occurred between 1920 and 1970)*1.

A measuring station of the local sea level in Venice wasn’t installed until 1897, so that the sea level of the previous centuries had to be estimated. It was determined with the help of the paintings by Veronese (i.e. 1571) and Canaletto (i.e. 1727–1758) and by measuring the submersion of the water stairs of the historic palaces along the Grand Canal.*2

In order to simplify the highly complex topic and its strong fluctuations, we opted for Roy Butterfield's linear fitting estimate of Venice’s Relative Sea-Level change of 1.43mm/year (without taking climate change into account) and put a Sea-Level model of the IPCC on top of it.

Performance evaluation

One of the foremost challenges in the project was to select our performance evaluation criteria: since the goal is to render paintings and drawings photorealistic, we had to rely on our subjective judgement of what we consider to be photorealistic.

We chose to focus on the following criteria to evaluate the performances of our model:

• Appropriate coloration according to reality: correctly colored buildings, no over-saturation for a single color.
• Consistent structures: buildings are not removed, and the structures are not modified too much, mainly in enhancing or reducing detail to make the image more realistic.
• Not overfitting to noise: sometimes paintings can hold a great amount of detail in brush strokes, for instance in clouds, and the GAN model tends to interpret some of these details as structure details, whereas such noise should be ignored most of the time.

We can see in Fig. 4 and Fig. 5 how we observed bad performances in the models while tuning parameters, according to these evaluation criteria. In Fig. 4, we can see how the low Canny filter parameters caused the over-representation of the colour red from the input image in the output image. In Fig. 5, we see how the model overfit to the noise in the input image, transforming a cloud into a building.

Fig. 4: Low canny filters, over-representing the red hues of the image

Fig. 5: Cloud transformed into a building, due to high learning rate/high relative number of learning epochs

Challenges (TODO remove)

Data collection

• Need more data than expected (black and white drawings, black and white paintings, coloured paintings, underwater data, Venice photos, Venice landmark photos)
• Need to clean more data than expected (removing watermarks, borders, removing unusual images)
• Content in input and output images does not always match. For example:
  • Large dense crowds in paintings that do not appear in modern-day photos
  • Old boat structures vs new boat structures
• Collecting good quality underwater data that contains details of underwater structures, rather than just images of algae and pond scum 🤿

Model challenges

• Long training time (~20 hours for 400 epochs on one GPU)
• Many hyperparameters to tune: we must carefully select dependent hyperparameters to test model performance on each sweep
• The CUT model requires more images than described in the paper

Quality Assessment

TODO: show results of the models

Limitations

We acknowledge several limitations of our methods, mainly caused by incorrect assumptions and time constraints.

One incorrect assumption that was made in the start of the project was that we would be able to see differences in model performances using quantitative measures, such as generator loss. In the initial test of the edge detection component, we used sweeps to observe generator and NCE loss of different models, to choose how to select the optimal models. However, after initial tests, we observed that the quantitative loss metrics did not vary drastically between the various models, whereas the outputs were significantly different. We can see the loss functions of four trained models in Fig. 6, compared to each of their outputs for the same input image in Fig. 7. We thus continued to test models by manually evaluating their results, considering factors described in the previous section.

Another limitation to our project was that in addition to the many preexisting hyperparameters of the CUT model, we added multiple extensions, which made the number of parameters to test unreasonable in the time frame that we had. We tested as many hyperparameter configurations as we could in the time frame given, occasionally causing us to delay our planning, but if we had a larger time frame we could further optimize the model performance by testing more configurations with a larger number of training epochs.

Fig. 6: Loss metrics (visualized using WandB)

Fig. 7: Model outputs for the four models trained in Fig. 6 (cp1 to cp4 from left to right)

Further work

In a further work, one could test more combinations of hyperparameters for our models, in addition to using larger and more images for the training. This model could also be used to create photorealistic renderings from paintings of other cities, such as Paris or Rome, although it would have to be retrained on images of these cities. We hope that our project will allow others to see historical portrayals of Venice in a more modern way, to better understand how Venice looked in its earlier days.

Links

• Contrastive Unpaired Translation GitHub
• Canny edge detector (for edge loss implementation)