Как показать изображение из сжатого представления (RNN)? - PullRequest
Новая
       40

Как показать изображение из сжатого представления (RNN)?

0 голосов
/ 19 апреля 2020

Я изучаю RNN с pytorch из этого github.

Pytorch RNN Tutorial

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

Я пытаюсь сохранить изображение из вывода, но в результате я получаю сжатое представление (ниже изображение, полученное из вывода модели)

compressed representation

# Train the model
total_step = len(train_loader)
for epoch in range(num_epochs):
    for i, (images, labels) in enumerate(train_loader):
        images = images.reshape(-1, sequence_length, input_size).to(device)
        labels = labels.to(device)

        # Forward pass
        outputs = model(images)
        loss = criterion(outputs, labels)

        # Backward and optimize
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()

        if (i+1) % 100 == 0:
            print ('Epoch [{}/{}], Step [{}/{}], Loss: {:.4f}' 
                    .format(epoch+1, num_epochs, i+1, total_step, loss.item()))

Я пытаюсь показать изображение вывода, добавив этот код 

plt.imshow(outputs.reshape(28, 28))

Но я получил эту ошибку 

RuntimeError: shape '[28, 28]' is invalid for input of size 1000

Итак, я предполагаю, что на выходе будет размер 1000. но я хочу, чтобы оно отображалось в виде изображения mnist (размер 784 (28 * 28)).

...