训练集与测试机的 accuracy: 0.9946，但是检测出图像灰度为全为1

[hp-93]

Describe the bug
A clear and concise description of what the bug is.
训练集与测试机的 accuracy: 0.9946，但是检测出图像为全为1，感觉np.reshape不对，跟onehot 操作不对应
images_test = np.reshape(np.argmax(y_test, axis=-1), (-1, image_h_w, image_h_w))
To Reproduce
Steps to reproduce the behavior:

1. Go to '...'
2. Click on '....'
3. Scroll down to '....'
4. See error

Expected behavior
A clear and concise description of what you expected to happen.

Screenshots
If applicable, add screenshots to help explain your problem.

Desktop (please complete the following information):

• OS: [e.g. iOS]
• Browser [e.g. chrome, safari]
• Version [e.g. 22]

Smartphone (please complete the following information):

• Device: [e.g. iPhone6]
• OS: [e.g. iOS8.1]
• Browser [e.g. stock browser, safari]
• Version [e.g. 22]

Additional context
Add any other context about the problem here.

[github-actions]

Thank you very much. This is your first issue.

[KakiWei]

同问，请问解决了吗

[ZiaoZhang1]

你好，请问data能发一下网盘链接吗，作者的数据集链接失效了