media-comp/2022-adain-pytorch

2022-adain-pytorch's Issues

User Interface

Build a user interface that allows users to interact with this project more easily. Feel free to use any forms of UI, such as a website or Python GUI. The program should support the following functionalities.

• Allow users to upload images path/directory and specify other arguments
• Perform style transfer tasks in test.py, test_interpolate.py, and test_video.py
• Display the output

Preserve Color

To preserve the color in the content image, first match the color distribution of the style image to that of content image, then perform a normal style transfer in test.py. Some simple color transfer methods include Histogram matching, mean/variance matching, linear transforms, etc. Feel free to use any methods

Spatial Control

Transfer different regions of content image to different styles. Perform AdaIN style transfer in test.py separately to different regions in the content feature maps using different style feature maps. Assign proper masking to different regions of the content image.