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?
zwx8981 commented
@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.