AncTe/ginka-generator
1
0
Fork 0
mirror of https://github.com/unanmed/ginka-generator.git synced 2026-05-22 02:44:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: 调整 heatmap 模型参数

Browse Source
This commit is contained in:
unanmed 2026-04-05 22:16:01 +08:00
parent b4f49e702d
commit 15cf564c2e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ginka/train_heatmap.py
View File

@ -26,8 +26,8 @@ from .utils import nms_sampling
# 0. 墙壁热力图, 1. 怪物热力图, 2. 资源热力图, 3. 血瓶热力图, 4. 宝石热力图, 5. 钥匙热力图 # 0. 墙壁热力图, 1. 怪物热力图, 2. 资源热力图, 3. 血瓶热力图, 4. 宝石热力图, 5. 钥匙热力图
# 6. 道具热力图, 7. 入口热力图, 8. 门热力图 # 6. 道具热力图, 7. 入口热力图, 8. 门热力图
BATCH_SIZE = 8 BATCH_SIZE = 128
VAL_BATCH_DIVIDER = 8 VAL_BATCH_DIVIDER = 64
NUM_CLASSES = 16 NUM_CLASSES = 16
MASK_TOKEN = 15 MASK_TOKEN = 15
GENERATE_STEP = 8 GENERATE_STEP = 8