Xiaoqi-Zhao-DLUT / GateNet-RGB-Saliency

(IJCV 2024&ECCV 2020 Oral) Suppress and Balance: A Simple Gated Network for Salient Object Detection
MIT License
162 stars 15 forks source link

There is a mistake in repoducing generate_ visfeamaps.py. #10

Closed hnn888 closed 3 years ago

hnn888 commented 3 years ago

Thanks for the valuable contribution! I'm reproducing generate_ visfeamaps.py, there is a mistake. I don't know how to correct it. Could you help me to answer it? The error is as follows: output1,output2,output3,output4 = net(img_var) ValueError : not enough values to unpack (expected 4, got 1)

I have seen the GateNet network structure, and there is only one output, I modify the code as follows output = net(img_var) And I got only one graph, like a heat map, not salient map.