Open HAOCHENYE opened 3 years ago
Hi, I found that the generator(model) have 4 modules that have used the depth map. I do not know which is the depth branch.
I have tried to change the input channel of the 4 modules to make the depth map be useless. It can train before 10 epochs, then it gets the nan params. It confuses me very much. Please tell me more detail. Thank you.
the current code is for rgbd saliency. you can remove the depth branch and train with rgd saliency training dataset, and then test with the rgb image.
On Mon, 25 Jan 2021 at 22:45, Mashiro notifications@github.com wrote:
-- Jing Zhang Ph.D. Student College of Engineering and Computer Science, Australian National University. Email: zjnwpu@gmail.com