Skip to content
This repository has been archived by the owner on Jul 2, 2023. It is now read-only.

About VGG16 pretrained model #52

Closed
zijundeng opened this issue Aug 29, 2017 · 6 comments
Closed

About VGG16 pretrained model #52

zijundeng opened this issue Aug 29, 2017 · 6 comments
Assignees
Labels

Comments

@zijundeng
Copy link

Hi @wkentaro , I see that you use the vgg16 pretrained model from caffe. I wonder that why you do not use a pretrained one provided by torchvision. Is the pretrained one from caffe better?

Thanks.

@wkentaro
Copy link
Owner

wkentaro commented Sep 5, 2017

I wanted to reproduce the same result as the original paper, and since pytorch uses different implementation for vgg (pytorch normalizes pixel value from 0 to 1), I decided to copy weights from caffe.
I have not trained fcn using vgg model in torchvision.
Does the model achieve the same result as the original vgg?

@wkentaro wkentaro self-assigned this Sep 5, 2017
@zijundeng
Copy link
Author

I have tried to use the pretrained vgg from torchvision for several times but failed to achieve the same result.

@zijundeng
Copy link
Author

The mean iu with the pretrained vgg from torchvision is 60~62. Maybe I haven't found a good set of hyperparameters.

@wkentaro
Copy link
Owner

wkentaro commented Sep 23, 2017

It seems that this issue is not the problem of pytorch-fcn in actual.
I think the torchvision vgg does not achieve the same result on imagenet, either. If I am wrong, please report.

Please request me to reopen this if you need.

@dillondavis
Copy link

@zijundeng Did you ever find good hyperparameters for the torchvision vgg16? I'm having some issues training it

@wkentaro wkentaro pinned this issue Sep 4, 2019
@hylovezhang
Copy link

Can you help me ? why i use the vgg16_from caffe model to train , the result of testing is zero? @zijundeng

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
Projects
None yet
Development

No branches or pull requests

4 participants