search

Karel911 / TRACER

TRACER: Extreme Attention Guided Salient Object Tracing Network (AAAI 2022) implementation in PyTorch
Apache License 2.0
195 stars 41 forks source link

Accessing The Actual Predicited Masks #5

Closed malcolmagius19 closed 2 years ago

malcolmagius19 commented 2 years ago

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 commented 2 years ago

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 commented 2 years ago

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