Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

About I3D model for VFID score. #42

Closed
kyuyeonpooh opened this issue Dec 4, 2020 · 0 comments
Closed

About I3D model for VFID score. #42

kyuyeonpooh opened this issue Dec 4, 2020 · 0 comments

Comments

@kyuyeonpooh
Copy link

Hi, thanks for sharing great codes!

I have a question about the mode (training / testing) you have used for I3D networks for evaluating VFID score.

def init_i3d_model():
global i3d_model
if i3d_model is not None:
return
logger.info("Loading I3D model for FID score ..")
i3d_model_weight = '../libs/model_weights/i3d_rgb_imagenet.pt'
if not os.path.exists(i3d_model_weight):
make_dirs(os.path.dirname(i3d_model_weight))
urllib.request.urlretrieve('https://github.com/piergiaj/pytorch-i3d/'
'raw/master/models/rgb_imagenet.pt', i3d_model_weight)
i3d_model = InceptionI3d(400, in_channels=3, final_endpoint='Logits')
i3d_model.load_state_dict(torch.load(i3d_model_weight))
i3d_model.to(torch.device('cuda:0'))

As shown above in I3D model initialization part for I3D feature extracting, it seems that you haven't converted the model into evaluation mode by i3d_model.eval().

When I evaluate your model without converting I3D into eval mode, I was able to reproduce the similar results in the paper.
However, when I turned the model into evaluation mode (because using .eval() is common for using the model as feature extractor), I found the VFID score goes up to over 1.0.

Is there any reason why the I3D network is used as training mode?
Thank you for reading! Have a nice day.

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

No branches or pull requests

1 participant