NVIDIA/DALI

Question on video loading

Open

#3,907 opened on May 16, 2022

View on GitHub
 (14 comments) (0 reactions) (1 assignee)C++ (670 forks)auto 404
PyTorch LightningVideohelp wanted

Repository metrics

Stars
 (5,722 stars)
PR merge metrics
 (PR metrics pending)

Description

@JanuszL and team hope you're doing OK. I have a pytorch video model that assumes each batch element is a five-frame video sequence around a frame at index ind, i.e., [ind-2, ind-1, ind, ind+1, ind+2]. Now, I need to pass a batch of such sequences to that model, ie.,

seqs = []
for ind in range(batch_size):
    seq = # images [ind-2, ind-1, ind, ind+1, ind+2]
    seqs.append(seq)

I have a vision of how to do this in pytorch after getting the sequence from DALI. is there a simpler way to do this in DALI?

Contributor guide