Googolxx/STF

specific path

Closed this issue · 5 comments

lunalovegood600 commented

Hi !
Thank you for the great work ! could you please give me the specific path，e.g. the specific path of the evaluate command ：-r reconstruction path -p path of checkpoint？thank u very much！

Googolxx commented

In the "STF" directory, running "mkdir reconstruction ckpt", then downloading checkpoint to the directory "ckpt".
Running"CUDA_VISIBLE_DEVICES=0 python -m compressai.utils.eval_model -d /path/to/image/folder/ -r reconstruction -a stf -p ckpt --cuda"

lunalovegood600 commented

thanks for your reply，I use the command "python downloader_openimages.py"to download the dataset "openimages" in the folder STF. Then I use the command"CUDA_VISIBLE_DEVICES=0 python -m compressai.utils.eval_model -d /openimages/test/data/ -r reconstruction -a cnn -p ckpt --cuda",but why there is an error "No such file or dictionary:'/openimages/test/data/'" ? Is there any fault in my path of the dataset？Or could u please give me the specific path of the datasets？I would really appreciate it if you could give me some answers！

Googolxx commented

Using “openimages/test/data/” instead of "/openimages/test/data/"

lunalovegood600 commented

thanks a lot！By the way，could u please tell me where can I download checkpoint？ The checkpoint file is the cnn_xxxx_best.pth ?Please forgive me if I'm getting this wrong.

Googolxx commented

Yes, the checkpoint file is "cnn_xxxx_best.pth" , Download links have been provided in "Readme.md". For example, cnn_0035_best.pth