Is there any other operation that needs to be noted if the result of my run is incorrect

This is the image I output according to the code. Do I need any further operations to output the image in the paper


<img width="959" alt="Image" src="https://github.com/user-attachments/assets/60d9e5b7-e0e0-43ce-9e7f-0b3acb3a63a8" />

The image in the paper is this：

<img width="489" alt="Image" src="https://github.com/user-attachments/assets/a69d802a-dc63-4698-9389-6c4dfc6aa663" />
