GitHub - aglooney/senior_thesis_final

Video Captioning baseline implementation using Pytorch

Steps

preprocess videos and labels python prepro_feats.py --output_dir data/feats/resnet152 --model resnet152 --n_frame_steps 40 --gpu 4,5 python prepro_vocab.py
Training a model python train.py --gpu 0 --epochs 3001 --batch_size 300 --checkpoint_path data/save --feats_dir data/feats/resnet152 --model S2VTAttModel --with_c3d 1 --c3d_feats_dir data/feats/c3d_feats --dim_vid 4096
Test

python eval.py --recover_opt data/save/opt_info.json --saved_model data/save/model_1000.pth --batch_size 100 --gpu 1

Name		Name	Last commit message	Last commit date
MSR-VTT/data_preprocess		MSR-VTT/data_preprocess
MSVD		MSVD
coco-caption		coco-caption
misc		misc
models		models
README.md		README.md
basic_vid_stats.py		basic_vid_stats.py
c3dv3.py		c3dv3.py
dataloader.py		dataloader.py
eval.py		eval.py
i3d.py		i3d.py
multiple_transfer_learning.py		multiple_transfer_learning.py
num_vids_in_dir.py		num_vids_in_dir.py
opts.py		opts.py
plot_transfer_learning.py		plot_transfer_learning.py
prepro_audio_v2.py		prepro_audio_v2.py
prepro_feats.py		prepro_feats.py
prepro_feats_v2.py		prepro_feats_v2.py
prepro_ngrams.py		prepro_ngrams.py
prepro_testset.py		prepro_testset.py
prepro_vocab.py		prepro_vocab.py
pytorch_i3d.py		pytorch_i3d.py
sys_opns.py		sys_opns.py
train.py		train.py
transfer_learning.py		transfer_learning.py

Navigation Menu