Skip to content
Snippets Groups Projects
Commit 47fc89ef authored by Kirk Ford's avatar Kirk Ford
Browse files

Update README.md

parent 0d3d6480
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment