zziang commited on
Commit
ed3b86e
·
verified ·
1 Parent(s): 9aabf6a

Upload encoder/code/k3_sheet.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. encoder/code/k3_sheet.py +10 -0
encoder/code/k3_sheet.py CHANGED
@@ -38,6 +38,12 @@ OUT.mkdir(parents=True, exist_ok=True)
38
  # 262144 -> 全部 8192 点。浏览器按 65536 做预算,教师要证明"原厂水平"就得给满。
39
  N_GAUSS = int(os.environ.get("FLIP_SHEET_GAUSS", "65536"))
40
 
 
 
 
 
 
 
41
  os.environ.setdefault("FLIP_WORK", str(WORK))
42
 
43
 
@@ -97,6 +103,10 @@ def main():
97
  k3 = load_k3()
98
  sys.path.insert(0, str(k3.REPO))
99
  import triposplat as T # noqa: PLC0415
 
 
 
 
100
 
101
  imgs = sorted(q for q in IMGDIR.rglob("*")
102
  if q.suffix.lower() in (".png", ".jpg", ".jpeg", ".webp")
 
38
  # 262144 -> 全部 8192 点。浏览器按 65536 做预算,教师要证明"原厂水平"就得给满。
39
  N_GAUSS = int(os.environ.get("FLIP_SHEET_GAUSS", "65536"))
40
 
41
+ # 输入分辨率:upstream 的 preprocess_image 用模块级 _CANVAS_SIZE,改它会同时改
42
+ # 抠图/裁切/最终画布。DINO patch 数按 size/16 变:1024 -> 4101 token,
43
+ # 512 -> 1029 token。flow model 的 cond 是 concat + self-attn,长度不写死;
44
+ # 输出高斯数不受影响。只有显式设了 FLIP_SHEET_IN 才去动 upstream 的全局。
45
+ IN_SIZE = int(os.environ.get("FLIP_SHEET_IN", "1024"))
46
+
47
  os.environ.setdefault("FLIP_WORK", str(WORK))
48
 
49
 
 
103
  k3 = load_k3()
104
  sys.path.insert(0, str(k3.REPO))
105
  import triposplat as T # noqa: PLC0415
106
+ if IN_SIZE != 1024:
107
+ T._CANVAS_SIZE = IN_SIZE
108
+ p("input canvas patched ->", IN_SIZE, "(teacher DINO tokens ->",
109
+ (IN_SIZE // 16) ** 2 + 5, ")")
110
 
111
  imgs = sorted(q for q in IMGDIR.rglob("*")
112
  if q.suffix.lower() in (".png", ".jpg", ".jpeg", ".webp")