search

imatge-upc / salgan

SalGAN: Visual Saliency Prediction with Generative Adversarial Networks
https://imatge-upc.github.io/salgan
MIT License
370 stars 105 forks source link

producing saliency map for other images #35

Open ashusdce opened 6 years ago

ashusdce commented 6 years ago

I am naive in python and trying to produce the saliency maps for some other images. But I am not able to understand the command given for predicting saliency maps. THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32,lib.cnmem=1,optimizer_including=cudnn python 03-predict.py

Also when I try to run 03predict.py file then I get the following error: No module name 'generator'.

Please help me in reproducing the results.

eduardramon commented 6 years ago

It could be due to this: https://stackoverflow.com/questions/30249119/difference-in-package-importing-between-python-2-7-and-3-4

Basically, python2.7 and python3.x assume different base_paths when importing the modules. You're probably running the script in python3 instead of python2.7.

To verify it, just run in the terminal: python --version.

If python3.x appears as your default version just substitute 'python' by 'python2.7' when running the command.

If you installed the required packages for python3, you'll need to reinstall them for python2.7.

Hope it helps!