{
 "name": "HairDeck",
 "format": "safetensors (state dict; args, step and token dim as JSON in the metadata key hairdeck)",
 "license": "apache-2.0",
 "manifest": "MANIFEST.json",
 "models": {
  "card_vae": {
   "checkpoint": "ckpt/card_vae/model.safetensors",
   "params_million": 8.07
  },
  "point_cloud_dit": {
   "checkpoint": "ckpt/pointcloud_model/model.safetensors",
   "params_million": 181.3,
   "sampling": "scripts/test.py pointcloud  (the test set: 50 Euler steps, CFG 2, seeds 0, 1, 2 in results/pointcloud/seed<k>, then the evaluation)"
  },
  "chain_dit": {
   "checkpoint": "ckpt/chain_model/model.safetensors",
   "params_million": 30.5,
   "sampling": "scripts/test.py chains  (the test set on the artists' cards: 50 Euler steps, CFG 2, seeds 0, 1, 2 in results/chains/seed<k>, then the evaluation; --cards results/pointcloud: on the point-cloud model's cards, chain seed k on card seed k)"
  },
  "image_dit": {
   "checkpoint": "ckpt/image_model/model.safetensors",
   "about": "the image model: RGB input (DINOv3 ViT-L/16 tokens of the 448 px crop), head frame, widened synthetic augmentation, restyled renders mixed in with p 0.33, initialised from the point-cloud model, 100K steps",
   "params_million": 182.3,
   "sampling": "scripts/image_to_hair.py IMAGE --out OUT  (hairdeck.infer.image.Pipeline: Euler 50, CFG 2, seed 0)"
  },
  "color_dit": {
   "checkpoint": "ckpt/color_model/model.safetensors",
   "about": "the colour model: the image model's network fine-tuned to predict 16 CIELAB colours along every generated card from the 448 px crop and the person matte; trained on renders of the training hairstyles with the colours of their VRM materials, starting from the image model",
   "params_million": 183.5,
   "sampling": "scripts/image_to_hair.py IMAGE --out OUT  (hairdeck.infer.image.Pipeline: one pass on the clean card tokens of every variation)"
  }
 },
 "data": {
  "avatar_body": {
   "files": [
    "data/avatar/body.json",
    "data/avatar/avatar.vrm",
    "data/avatar/hair.glb"
   ],
   "about": "the avatar body of the VRM export (hairdeck.vrm; a VRoid Studio template avatar): the VRM, its original hair (hair.glb, for the check that the export removed it) and body.json with its head-frame fit and the VRM meta written into exported files (author hyz317, Apache-2.0)"
  },
  "demo_examples": {
   "files": [
    "data/demo_examples/examples.json",
    "data/demo_examples/ex01.png",
    "data/demo_examples/ex02.png",
    "data/demo_examples/ex03.png",
    "data/demo_examples/ex04.png",
    "data/demo_examples/ex05.png",
    "data/demo_examples/ex06.png",
    "data/demo_examples/ex07.png",
    "data/demo_examples/ex08.png",
    "data/demo_examples/ex09.png",
    "data/demo_examples/ex10.png",
    "data/demo_examples/ex11.png",
    "data/demo_examples/ex12.png"
   ],
   "about": "the 12 example images of the demo (examples.json lists them in the order shown)"
  },
  "test_data": {
   "files": [
    "data/test_cards_orig/",
    "data/test_cards_orig_dirs.npz",
    "data/spring/chains_test.jsonl",
    "data/spring/chains_test_head.npz",
    "data/cards_test.npz",
    "ckpt/card_vae/latents_test_head.npy",
    "data/head_frame_test.npz",
    "data/head_params_test.json"
   ],
   "about": "evaluation data of scripts/test.py (file names: test row, 000 to 099): the cards of the original hair meshes of the 100 test cases (CHARM-format card json, 8 columns; for the 24 combined-template cases, of their two template meshes) and the true width direction at each of their control points (test_cards_orig_dirs.npz), read by the pointcloud test; the spring-bone chains of the test characters (as extracted from their VRM files, and as chain tokens in the head frame) and the card features and head-frame card latents of the GT cards that the chain model is conditioned on, and the test characters' head fits and head frames, read by the chains test"
  },
  "inference_data": {
   "files": [
    "data/token_stats.json",
    "data/token_stats_head.json",
    "data/head_frame_train.npz",
    "data/spring/chain_stats_head.json"
   ],
   "about": "card token normalisation (point-cloud and head-frame models), the canonical head of the head frame (W, O; head proxy), chain token statistics of the chain DiT"
  }
 },
 "files": [
  "ckpt/card_vae/latent_stats.json",
  "ckpt/card_vae/latent_stats_head.json",
  "ckpt/card_vae/latents_test_head.npy",
  "ckpt/card_vae/model.safetensors",
  "ckpt/card_vae/stats.json",
  "ckpt/chain_model/model.safetensors",
  "ckpt/color_model/model.safetensors",
  "ckpt/image_model/model.safetensors",
  "ckpt/pointcloud_model/model.safetensors",
  "data/avatar/avatar.vrm",
  "data/avatar/body.json",
  "data/avatar/hair.glb",
  "data/cards_test.npz",
  "data/demo_examples/ex01.png",
  "data/demo_examples/ex02.png",
  "data/demo_examples/ex03.png",
  "data/demo_examples/ex04.png",
  "data/demo_examples/ex05.png",
  "data/demo_examples/ex06.png",
  "data/demo_examples/ex07.png",
  "data/demo_examples/ex08.png",
  "data/demo_examples/ex09.png",
  "data/demo_examples/ex10.png",
  "data/demo_examples/ex11.png",
  "data/demo_examples/ex12.png",
  "data/demo_examples/examples.json",
  "data/head_frame_test.npz",
  "data/head_frame_train.npz",
  "data/head_params_test.json",
  "data/spring/chain_stats_head.json",
  "data/spring/chains_test.jsonl",
  "data/spring/chains_test_head.npz",
  "data/test_cards_orig/000.json",
  "data/test_cards_orig/001.json",
  "data/test_cards_orig/002.json",
  "data/test_cards_orig/003.json",
  "data/test_cards_orig/004.json",
  "data/test_cards_orig/005.json",
  "data/test_cards_orig/006.json",
  "data/test_cards_orig/007.json",
  "data/test_cards_orig/008.json",
  "data/test_cards_orig/009.json",
  "data/test_cards_orig/010.json",
  "data/test_cards_orig/011.json",
  "data/test_cards_orig/012.json",
  "data/test_cards_orig/013.json",
  "data/test_cards_orig/014.json",
  "data/test_cards_orig/015.json",
  "data/test_cards_orig/016.json",
  "data/test_cards_orig/017.json",
  "data/test_cards_orig/018.json",
  "data/test_cards_orig/019.json",
  "data/test_cards_orig/020.json",
  "data/test_cards_orig/021.json",
  "data/test_cards_orig/022.json",
  "data/test_cards_orig/023.json",
  "data/test_cards_orig/024.json",
  "data/test_cards_orig/025.json",
  "data/test_cards_orig/026.json",
  "data/test_cards_orig/027.json",
  "data/test_cards_orig/028.json",
  "data/test_cards_orig/029.json",
  "data/test_cards_orig/030.json",
  "data/test_cards_orig/031.json",
  "data/test_cards_orig/032.json",
  "data/test_cards_orig/033.json",
  "data/test_cards_orig/034.json",
  "data/test_cards_orig/035.json",
  "data/test_cards_orig/036.json",
  "data/test_cards_orig/037.json",
  "data/test_cards_orig/038.json",
  "data/test_cards_orig/039.json",
  "data/test_cards_orig/040.json",
  "data/test_cards_orig/041.json",
  "data/test_cards_orig/042.json",
  "data/test_cards_orig/043.json",
  "data/test_cards_orig/044.json",
  "data/test_cards_orig/045.json",
  "data/test_cards_orig/046.json",
  "data/test_cards_orig/047.json",
  "data/test_cards_orig/048.json",
  "data/test_cards_orig/049.json",
  "data/test_cards_orig/050.json",
  "data/test_cards_orig/051.json",
  "data/test_cards_orig/052.json",
  "data/test_cards_orig/053.json",
  "data/test_cards_orig/054.json",
  "data/test_cards_orig/055.json",
  "data/test_cards_orig/056.json",
  "data/test_cards_orig/057.json",
  "data/test_cards_orig/058.json",
  "data/test_cards_orig/059.json",
  "data/test_cards_orig/060.json",
  "data/test_cards_orig/061.json",
  "data/test_cards_orig/062.json",
  "data/test_cards_orig/063.json",
  "data/test_cards_orig/064.json",
  "data/test_cards_orig/065.json",
  "data/test_cards_orig/066.json",
  "data/test_cards_orig/067.json",
  "data/test_cards_orig/068.json",
  "data/test_cards_orig/069.json",
  "data/test_cards_orig/070.json",
  "data/test_cards_orig/071.json",
  "data/test_cards_orig/072.json",
  "data/test_cards_orig/073.json",
  "data/test_cards_orig/074.json",
  "data/test_cards_orig/075.json",
  "data/test_cards_orig/076.json",
  "data/test_cards_orig/077.json",
  "data/test_cards_orig/078.json",
  "data/test_cards_orig/079.json",
  "data/test_cards_orig/080.json",
  "data/test_cards_orig/081.json",
  "data/test_cards_orig/082.json",
  "data/test_cards_orig/083.json",
  "data/test_cards_orig/084.json",
  "data/test_cards_orig/085.json",
  "data/test_cards_orig/086.json",
  "data/test_cards_orig/087.json",
  "data/test_cards_orig/088.json",
  "data/test_cards_orig/089.json",
  "data/test_cards_orig/090.json",
  "data/test_cards_orig/091.json",
  "data/test_cards_orig/092.json",
  "data/test_cards_orig/093.json",
  "data/test_cards_orig/094.json",
  "data/test_cards_orig/095.json",
  "data/test_cards_orig/096.json",
  "data/test_cards_orig/097.json",
  "data/test_cards_orig/098.json",
  "data/test_cards_orig/099.json",
  "data/test_cards_orig_dirs.npz",
  "data/token_stats.json",
  "data/token_stats_head.json"
 ]
}