У меня есть набор данных видео. Я разделил каждое видео на набор фреймов, сохранил каждое видео в список и сохранил этот список в массиве, так что теперь у меня есть массив списков, каждый список - это видео, как я могу передать каждое видео всеть и как я могу установить длину последовательности для списка каждого видео
подсказка: каждое видео имеет разную длину
это мой код RNN
# Create RNN Model
class RNNModel(nn.Module):
def __init__(self, input_dim, hidden_dim, layer_dim,
output_dim):
super(RNNModel, self).__init__()
# Number of hidden dimensions
self.hidden_dim = hidden_dim
# Number of hidden layers
self.layer_dim = layer_dim
# RNN
self.rnn = nn.RNN(input_dim, hidden_dim, layer_dim,
batch_first=True,
nonlinearity='relu')
# Readout layer
self.fc = nn.Linear(hidden_dim, output_dim)
def forward(self, x):
# Initialize hidden state with zeros
h0 = Variable(torch.zeros(self.layer_dim, x.size(0),
self.hidden_dim))
# One time step
out, hn = self.rnn(x, h0)
out = self.fc(out[:, -1, :])
return out