Add color transfer as in paper

mblondel / smooth-ot

Python implementation of smooth optimal transport.

BSD 2-Clause "Simplified" License

54 stars 13 forks source link

Closed cptq closed 3 years ago

cptq commented 3 years ago

Adds code and a script for computing and plotting color transfer as in the paper. Instructions are in color-transfer/README.md.

mblondel commented 3 years ago

Thanks a lot Derek!

Could you

I'll merge after that! Thanks a lot :)

cptq commented 3 years ago

Hi @mblondel thanks for taking a look. I think I made the changes you wanted.

mblondel commented 3 years ago

Two other comments before merging:

If you use root_dir = os.path.dirname(os.path.abspath(__file__)), you will get the directory where the script (dataset.py or plot_color_transfer.py) is, regardless where it's executed from. You can then use, e.g., os.path.join(root_dir, "..", "..", "data", "autumn.jpg") to access particular files. I should have caught this before, sorry.
the README file should be moved to the color-transfer folder

Thanks in advance.

cptq commented 3 years ago

I see, thanks for the tips, how does this look @mblondel ?

mblondel commented 3 years ago

Thanks a ton @cptq! Looking forward to see what comes out of your research!