subpic / ava-mlsp

MLSP feature learning for the AVA aesthetics database
MIT License
76 stars 15 forks source link

which version you use in this project? #1

Closed flyingbird93 closed 5 years ago

flyingbird93 commented 5 years ago

hi, I see your study in aesthetic assessment, it`s so exciting to improve the SROCC and LCC to 0.75. I want to know which version you use keras and tensorflow, could your please write a requirements for us to apply your results quickly.

subpic commented 5 years ago

Hi,

The environment I'm running the code under is:

Python 2.7.16 :: Anaconda, Inc 
tensorflow-gpu 1.14.0
keras-gpu 2.2.4

The code runs on google colab as well, see notebook example here. The training itself is slower on colab due to slower disk reads, than when using your own SSD.

Alternatively you can deploy the code via docker using the Python 2 environment in jupyter-data-science.

Best, Vlad