pytorch/examples

No saved *.png in checkpoint in dcgan.cpp

Open

#742 aberto em 24 de mar. de 2020

Ver no GitHub
 (3 comments) (0 reactions) (0 assignees)Python (21.634 stars) (9.429 forks)batch import
bughelp wanted

Description

In the README file of "DCGAN Example with the PyTorch C++ Frontend" it says that:

The training script periodically generates image samples. Use the display_samples.py script situated in this folder to generate a plot image. For example:

But the dcgan.cpp file just saves the model in *.pt format, doesnt save any picture. Then, the command stated in the README:

$ python display_samples.py -i dcgan-sample-10.png
Saved out.png

Gives an error as dcgan-sample-10.png doesnt exist

Guia do colaborador

Pilha de tecnologia
cpppython
Domain
machine learning
Tipo Issue
bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 hours
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
stale
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
clear
Prerequisites
C++ basicsPyTorch C++ API basics
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
70
Direção de pesquisa
The issue is in the dcgan.cpp file within the pytorch/examples repository. The README states that sample images are periodically saved as PNG, but the C++ code only saves model checkpoints. To fix, one must add image saving logic to dcgan.cpp, possibly using stb image write or a similar library. The display samples.py script expects PNG files; ensure the generated images are saved in the correct format and location.