From 5df14a3f3f1f9d7206942d1531b75a33d5d1f527 Mon Sep 17 00:00:00 2001
From: unanmed <1319491857@qq.com>
Date: Tue, 7 Apr 2026 22:57:52 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20heatmap=20=E9=AA=8C=E8=AF=81=E7=B4=A2?=
 =?UTF-8?q?=E5=BC=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ginka/train_heatmap.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ginka/train_heatmap.py b/ginka/train_heatmap.py
index 81d9601..46a213f 100644
--- a/ginka/train_heatmap.py
+++ b/ginka/train_heatmap.py
@@ -195,6 +195,8 @@ def train():
                         generated_img = matrix_to_image_cv(map.view(B, H, W)[0].cpu().numpy(), tile_dict)
                         cv2.imwrite(f"result/final_img/{idx}.png", generated_img)
                     
+                    idx += 1
+                    
                 # 3. 完全随机生成五张图
                 if args.use_maskgit:
                     for i in range(0, 5):