unconditional_demo.py error

[chris901003]

mmgeneration/demo/unconditional_demo.py

Line 72 in 92be69b

results = (results[:, [2, 1, 0]] + 1.) / 2.

Error when producing grayscale image, change it to the follow way to fix.

if results.shape[1] == 3:
    results = (results[:, [2, 1, 0]] + 1.) / 2.
elif results.shape[1] == 1:
    results = (results[:, [0]] + 1.) / 2.

[LeoXing1996]

@chris901003 Thanks for your interest in our project.
Would you like to create a Pull Request to fix this problem?

To create a Pull Request, you should refer this guidance.

[zengyh1900]

Please check this issue. @LeoXing1996 do you have the plan to fix this issue or set it in Backlog?