NVIDIA/DALI

Question on video loading

Open

#3.907 aberto em 16 de mai. de 2022

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

Métricas do repositório

Stars
 (5.722 stars)
Métricas de merge de PR
 (Métricas PR pendentes)

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?

Guia do colaborador