Update README.md
Browse files
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 |
-
|
|
|
|
|
|
|
| 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:
|