zwx8981/DBCNN-PyTorch

The pretrained SCNN

teowu opened this issue · 1 comments

teowu commented

Hi author:
I would like to make use on the pretrained SCNN for the further training.
However, I found out the pretrained 'scnn.pkl' file in this repo is only a string of numbers instead of model weights.
Where can I find the Pytorch weights for SCNN, or will this be publically available?

@TimothyHTimothy scnn.pkl is the model weights, you should first claim a scnn object, and then load the weights using " scnn.load_state_dict(torch.load(scnn_root))". See Line 52 in the DBCNN.py.