haoningwu commited on
Commit
311d3fa
·
verified ·
1 Parent(s): b81bc1f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -2
README.md CHANGED
@@ -118,15 +118,21 @@ bash scripts/train.sh
118
  ```
119
 
120
  ## 🧪 Evaluation
121
- To generate the 3D scenes on the 3D-FUTURE test set:
122
  ```
123
  bash scenegen_eval.sh
124
  ```
125
- To evaluate the trained model on the 3D-FUTURE test set:
 
 
126
  ```
127
  cd evalscene
128
  bash eval_scenegen.sh
129
  ```
 
 
 
 
130
 
131
  ## 📜 Citation
132
  If you use this code and data for your research or project, please cite:
 
118
  ```
119
 
120
  ## 🧪 Evaluation
121
+ To generate the 3D scenes on the 3D-FUTURE test set using the SceneGen model, use the following command:
122
  ```
123
  bash scenegen_eval.sh
124
  ```
125
+ which will use the `scenegen_eval.py` script to generate the normalized scenes.
126
+
127
+ To evaluate the trained SceneGen model on the 3D-FUTURE test set, use the following command:
128
  ```
129
  cd evalscene
130
  bash eval_scenegen.sh
131
  ```
132
+ Make sure to have the processed 3D-FUTURE dataset and the rendered images in place as described in the Dataset section and the evaluation configs in `evalscene/configs/test/scene_evaluation_scenegen.yaml` set correctly. Then the evaluation script will compute metrics between the normalized generated scenes and the ground truth.
133
+
134
+ Some packages used in the evaluation require additional installation. Please install the packages: `torchmetrics`, `lpips`, `clip`, and `probreg` via pip.
135
+
136
 
137
  ## 📜 Citation
138
  If you use this code and data for your research or project, please cite: