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

which one is the checkpoint? #116

Open
tangyuelm opened this issue Jul 5, 2021 · 5 comments
Open

which one is the checkpoint? #116

tangyuelm opened this issue Jul 5, 2021 · 5 comments

Comments

@tangyuelm
Copy link

Thanks for your codes and model. I want to download the i3d model pre-trained on the Kinetics dataset but feel confused about the checkpoint. I followed the path in evaluate_sample.py
_CHECKPOINT_PATHS = {
'rgb': 'data/checkpoints/rgb_scratch/model.ckpt',
'rgb600': 'data/checkpoints/rgb_scratch_kin600/model.ckpt',
'flow': 'data/checkpoints/flow_scratch/model.ckpt',
'rgb_imagenet': 'data/checkpoints/rgb_imagenet/model.ckpt',
'flow_imagenet': 'data/checkpoints/flow_imagenet/model.ckpt',
}
I copied the checkpoints folder to a new path and tried to load it.
However, is seems that no such file or directory: 'checkpoints/rgb_scratch_kin600/model.ckpt'
I found 4 files under the folder rgb_scratch_kin600 but cannot find the model.ckpt. Could anyone help me?
Besides, can the checkpoint be loaded to model writing in Pytorch?

@tangyuelm
Copy link
Author

tangyuelm commented Jul 6, 2021

To my understand, the rgb_scratch_600 means the checkpoints for RGB frames pre-trained on Kinetics-600. What do the folder rgb_scratch and flow_scratch mean?
Besides, in each folder, there are four files called "checkpoints", "model.ckpt.data-00000-of-00001", "model.ckpt.index", "model.ckpt.meta", but there is no file called model.ckpt. Which one should I use?
I am not familiar with Tensorflow, so I am wondering how to load these checkpoints in Pytorch. Do you have the .pth version for the checkpoint pre-trained on Kinetics-400? I can only find the checkpoints pre-trained on ImageNet and Charades in your Pytorch codes from https://github.com/piergiaj/pytorch-i3d, but I believe Kinetics-400 is more commonly used for pre-training.

@Tortoise17
Copy link

I am also looking for same answer. Or any way to convert them to .h5 format?

@joaoluiscarreira
Copy link

joaoluiscarreira commented Sep 28, 2021 via email

@joaoluiscarreira
Copy link

joaoluiscarreira commented Sep 28, 2021 via email

@Tortoise17
Copy link

@joaoluiscarreira thank you so much. Is there any command line with which I can input the mp4 and get back the logits and features? or do I have to export flow and frames for the input in model to get these.

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

3 participants