ccraft dataset added to handle outfits made of separate garments

Dolorousrtur · Oct 24, 2024 · 9b7cc21 · 9b7cc21
1 parent 312fd82
commit 9b7cc21
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/Inference.ipynb b/Inference.ipynb
@@ -88,7 +88,7 @@
     "\n",
     "We currently support [SMPL](https://smpl.is.tue.mpg.de/) and [SMPL-X](https://smpl-x.is.tue.mpg.de/) body models.\n",
     "\n",
-    "Note: you will not be able to run LBS-initialized simulation with a body bodel that is different from the one you specify in this step."
+    "Note: you will not be able to run LBS-initialized simulation with a body model that is different from the one you specify in this step."
    ]
   },
   {
@@ -392,6 +392,7 @@
     "# to use our (legacy) SMPL pose sequence\n",
     "# sequence_loader = 'hood_pkl'\n",
     "\n",
+    "\n",
     "# ====================================================================================================\n",
     "\n",
     "\n",

diff --git a/configs/contourcraft.yaml b/configs/contourcraft.yaml
@@ -46,7 +46,7 @@ dataloader:
   batch_size: 1
   pyg_data: True
   dataset:
-    postcvpr:
+    ccraft:
       data_root: 'vto_dataset/smpl_parameters'
       body_model_root: 'body_models'
       model_type: 'smpl'
@@ -56,7 +56,7 @@ dataloader:
       split_path: 'datasplits/train.csv'
       pinned_verts: True
       random_betas: False
-      n_coarse_levels: 3
+      n_coarse_levels: 4
       separate_arms: True
       wholeseq: True
       noise_scale: 0

diff --git a/static/all_garments.png b/static/all_garments.png
diff --git a/utils/validation.py b/utils/validation.py
@@ -96,7 +96,7 @@ def create_postcvpr_one_sequence_dataloader(sequence_path: str, garment_name: st
     file_name, _ = os.path.splitext(file_name)
 
     if config is None:
-        config = 'postcvpr'
+        config = 'contourcraft'
 
 
     dataloader = create_one_sequence_dataloader(use_config=config, data_root=data_root, single_sequence_file=file_name,