pytorch/examples

No saved *.png in checkpoint in dcgan.cpp

Open

Aperta il 24 mar 2020

Vedi su GitHub
 (3 commenti) (0 reazioni) (0 assegnatari)Python (21.634 star) (9429 fork)batch import
bughelp wanted

Descrizione

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

Guida contributor

Tech stack
cpppython
Dominio
machine learning
Tipo issue
bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 hours
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
stale
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
clear
Prerequisiti
C++ basicsPyTorch C++ API basics
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
70
Direzione di ricerca
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.