Upload encoder/code/k3_sheet.py with huggingface_hub
Browse files- 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")
|