search

Ugness / PiCANet-Implementation

Pytorch Implementation of PiCANet: Learning Pixel-wise Contextual Attention for Saliency Detection
MIT License
178 stars 40 forks source link

--img Unrecognized arguments #11

Closed Max-ChenFei closed 5 years ago

Max-ChenFei commented 5 years ago

In Execution Example session, It is

python Image_test.py --model_dir models/state_dict/07261950/10epo_1000000step.ckpt --img test --cuda cpu Acutally, I see

parser.add_argument('-img', '--image_dir', help='Directory of your test_image ""folder""', default='../test') in image_test.py

--img is unrecognized arguments. Maybe you should correct the command in execution example.

Ugness commented 5 years ago

Oh, I see. I correct it. Thanks.