fix: heatmap 验证索引

2026-05-14 04:41:12 +08:00 · 2026-04-07 22:57:52 +08:00 · 2026-04-07 22:57:52 +08:00 · 5df14a3f3f
commit 5df14a3f3f
parent 3e898dc5ba
1 changed files with 2 additions and 0 deletions
--- 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):