Atze00 / MoViNet-pytorch

MoViNets PyTorch implementation: Mobile Video Networks for Efficient Video Recognition;
MIT License
258 stars 50 forks source link

transform for test video in movinet_tutorial.ipynb #7

Closed Billchen1993 closed 3 years ago

Billchen1993 commented 3 years ago

In movinet_tutorial.ipynb, video transform for test dataset should be transform_test rather than transform I guess. Namely, hmdb51_test = torchvision.datasets.HMDB51('video_data/', 'test_train_splits/', num_frames,frame_rate=5, step_between_clips = clip_steps, fold=1, train=False, transform=transform_test, num_workers=2)

Atze00 commented 3 years ago

thank you for the issue