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
162 stars 15 forks source link

generate_sal.py #14

Open zhuyusenCV opened 1 year ago

zhuyusenCV commented 1 year ago

When I'm done training on my own dataset，there was a problem while testing。Can you tell me why?

net = GateNet().cuda() TypeError: init() missing 2 required positional arguments: 'block1' and 'layers'

Jerry050512 commented 14 hours ago

Replace the code using:

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

Just like the train.py