DSS-pytorch
DSS-pytorch copied to clipboard
:star: PyTorch implement of Deeply Supervised Salient Object Detection with Short Connection
as title states
RuntimeError: Error(s) in loading state_dict for ModuleList: Missing key(s) in state_dict: "0.weight", "0.bias", "2.weight", "2.bias", "5.weight", "5.bias", "7.weight", "7.bias", "10.weight", "10.bias", "12.weight", "12.bias", "14.weight", "14.bias", "17.weight", "17.bias", "19.weight", "19.bias", "21.weight",...
could you offer the tensorflow code ? thanks
Have you compared the results with paper?
I train the DSS models, and test it. when i test it, i found that: the saliency object boundary have too much noise. do u have the same problem? maybe...
Sorry to bother you but I had another issue when I tried to train with my own data sets: `python main.py --mode='train' --train_path='data/images' --label_path='data/ground_truth_mask' --batch_size=8 --visdom=False` It seemed to run...
Hi, I am trying to train a module using my own datasets. I follow all those instructions including installing pytorch (pip3 install --no-cache-dir torchvision). However, when I try to start...