PyTorch LightningVideohelp wanted
Métriques du dépôt
- Stars
- (5 722 stars)
- Métriques de merge PR
- (Métriques PR en attente)
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?