AncTe/ginka-generator
1
0
Fork 0
mirror of https://github.com/unanmed/ginka-generator.git synced 2026-05-14 04:41:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: heatmap 验证索引

Browse Source
This commit is contained in:
unanmed 2026-04-07 22:57:52 +08:00
parent 3e898dc5ba
commit 5df14a3f3f
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ginka/train_heatmap.py
View File

@ -195,6 +195,8 @@ def train():
generated_img = matrix_to_image_cv(map.view(B, H, W)[0].cpu().numpy(), tile_dict) 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) cv2.imwrite(f"result/final_img/{idx}.png", generated_img)
idx += 1
# 3. 完全随机生成五张图 # 3. 完全随机生成五张图
if args.use_maskgit: if args.use_maskgit:
for i in range(0, 5): for i in range(0, 5):