diff --git a/README.md b/README.md index dc853eb494e893f09a6208bf0eb8e0784f81c739..9cec70c8db4d6ec69dda25173bbf5551f910ecdb 100644 --- a/README.md +++ b/README.md @@ -7,19 +7,20 @@ pip install opencv-python pip install matplotlib pip install scikit-image pip install torch -pip install torch-vision +pip install torchvision pip install tqdm ``` The baseline model is completely contained and requires no extra installation. ## Usage Main Model: Open a file explorer and navigate to the "Use Palettenet" folder. Open a terminal in the "Use Palettenet" folder. Drop any .jpg you like into the folder and rename it "image.jpg". Run the command "python PaletteNetUse.py" for the ablated model with no adversarial training (looks better) or run "python PaletteNetUseAdv.py" to try the model with adversarial training. -You will be greeted by 4 color pickers in succession, each color you pick represents a colour in your 4-color palette. The new file will then save as "pict256.png" after all of the pyplot figures are closed. +You will be greeted by 4 color pickers in succession, each color you pick represents a colour in your 4-color palette. The new file will then save as "pict256.png" in "Use PaletteNet" after all of the pyplot figures are closed. Baseline Model: Open a file explorer and navigate to the "Use Photo Recoloring" folder. Open the "Index.html" file with a web browser such as chrome, and follow the on screen instructions to add a photo. Clicking the circular coloured buttons at the bottom will allow you to change colors. When finished, click confirm and then wait for the image to generate. ## Authors and acknowledgment -The SSIM script I used was found here: https://stackoverflow.com/questions/71567315/how-to-get-the-ssim-comparison-score-between-two-images +The SSIM script I used was made by Nathancy, found here: https://stackoverflow.com/questions/71567315/how-to-get-the-ssim-comparison-score-between-two-images +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