kenshohara / video-classification-3d-cnn-pytorch

Video classification tools using 3D ResNet
MIT License
1.09k stars 260 forks source link

error occurs in dataset.py #51

Closed xuantengh closed 4 years ago

xuantengh commented 4 years ago

Traceback (most recent call last): File "main.py", line 57, in <module> result = classify_video('tmp', input_file, class_names, model, opt) File "/home/share/Highlight/code/video-classification-3d-cnn-pytorch/classify.py", line 24, in classify_video for i, (inputs, segments) in enumerate(data_loader): File "/home/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 623, in __next__ return self._process_next_batch(batch) File "/home/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 658, in _process_next_batch raise batch.exc_type(batch.exc_msg) RuntimeError: Traceback (most recent call last): File "/home/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 138, in _worker_loop samples = collate_fn([dataset[i] for i in batch_indices]) File "/home/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 138, in <listcomp> samples = collate_fn([dataset[i] for i in batch_indices]) File "/home/share/Highlight/code/video-classification-3d-cnn-pytorch/dataset.py", line 130, in __getitem__ clip = torch.stack(clip, 0).permute(1, 0, 2, 3) RuntimeError: expected a non-empty list of Tensors This issue occurs when I use the feature mode and resnet-34-kinetics.pth model. And I set the batch_size 16. So how could I solve this problem? Thank you very much! > <