search

Xiaoqi-Zhao-DLUT / GateNet-RGB-Saliency

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

python generate_sal.py #15

Open zhuyusenCV opened 1 year ago

zhuyusenCV commented 1 year ago

Hello, I created the same environment as you, but I still can't succeed when running the build salmap, he prompts me that I need to enter two parameters. Can you help me?

File "generate_sal.py", line 83, in main() File "generate_sal.py", line 41, in main net = GateNet().cuda() TypeError: init() missing 2 required positional arguments: 'block1' and 'layers'

m0ho commented 1 year ago

try this

net = GateNet(Bottleneck, [3,4,6,3]).cuda()