Accessing The Actual Predicited Masks

[malcolmagius19]

Hi, I tried to replicate your code and managed to replicate the test on the DUTS test set using TE0_0 pretrained model. However I would like to know how to get the actual masks predicted of the images after running the test. Thanks, Malcolm

[Karel911]

Hi,
If you want to get the prediction map (actual masks), add the arguments below.
python main.py test --exp_num 0 --arch 0 --img_size 320 --save_map True
You can find the maps from the directory named pred_map/
Thanks.

[malcolmagius19]

Hi Karel, thanks for the quick response. Your suggestion worked perfectly, Legend <3