pytorch/examples

No saved *.png in checkpoint in dcgan.cpp

Open

#742 建立於 2020年3月24日

在 GitHub 查看
 (3 留言) (0 反應) (0 負責人)Python (21,634 star) (9,429 fork)batch import
bughelp wanted

描述

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

貢獻者指南

技術棧
cpppython
領域
machine learning
議題類型
bug
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
2
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-3 hours
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
stale
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
clear
前置要求
C++ basicsPyTorch C++ API basics
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
70
研究方向
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.