tzzcl / PSOL

code repository of “Rethinking the Route Towards Weakly Supervised Object Localization” in CVPR 2020

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

python PSOL_inference.py --loc-model {$LOC_MODEL} --cls-model {$CLS_MODEL} {--ten-crop} giving error!!!!

Shiven004 opened this issue · comments

Hi Zhang,

For testing - python PSOL_inference.py --loc-model {$LOC_MODEL} --cls-model {$CLS_MODEL} {--ten-crop}
I am using this cmd-
_python PSOL_inference.py --loc-model vgg16 --cls-model vgg16 {--ten-crop}_

How to specify path of the model in this ?Getting this error:
_FileNotFoundError: [Errno 2] No such file or directory: 'vgg16loc.pth.tar'_

I have kepts the pre-trained model tar files in model directory.

Please guide on this!!!!

image

Folder structure is -
image

What is value should be passed {--ten-crop}?

@tzzcl - I have commented the piece of code that requires --ten-crop args. Now the command is working fine.