Update README.md

dongjunKu · web-flow · commit 9021a6b2ac43 · 2024-03-18T21:24:28.000+09:00
diff --git a/README.md b/README.md
@@ -30,9 +30,10 @@ ContactGen
 ├── train_diffusion.py
 ├── train_guidenet.py
 ├── utils.py
+├── visualize.py
 ```
 
-you can get contact_regions.json from https://github.com/sminchisescu-research/imar_vision_datasets_tools/blob/main/info/contact_regions.json
+you can get [contact_regions.json](https://github.com/sminchisescu-research/imar_vision_datasets_tools/blob/main/info/contact_regions.json) here
 
 ## Training
 training diffusion module
@@ -47,3 +48,9 @@ python train_guidenet.py
 ```shell
 python sample.py
 ```
+It will generate sample in the ``output_diffusion_epoch1000_ci3d``
+
+## Visualizing
+```shell
+python visualize.py output_diffusion_epoch1000_ci3d/???_human_pred.pkl
+```
