Передача видео в RNN - PullRequest
0 голосов
/ 05 марта 2019

У меня есть набор данных видео. Я разделил каждое видео на набор фреймов, сохранил каждое видео в список и сохранил этот список в массиве, так что теперь у меня есть массив списков, каждый список - это видео, как я могу передать каждое видео всеть и как я могу установить длину последовательности для списка каждого видео

подсказка: каждое видео имеет разную длину

это мой код 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
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...