diff --git a/README.md b/README.md index 9cec70c8db4d6ec69dda25173bbf5551f910ecdb..dda7fbd049843c8d2058ef45dbd51d2582439e1e 100644 --- a/README.md +++ b/README.md @@ -23,5 +23,7 @@ The SSIM script I used was made by Nathancy, found here: https://stackoverflow.c Thank you for the public post. This project would not have been possible without the use/accessibility of the following models: + My main model is a modified version of Palettenet: https://github.com/yongzx/PaletteNet-PyTorch + My baseline is a slightly modified version of palette-based Photo Recoloring: https://github.com/b-z/photo_recoloring