Skip to content

Commit de6c27e

Browse files
committed
on-the-fly evaluation metrics script for model test is enables, along with readme evaluation update
1 parent 2029acb commit de6c27e

File tree

5 files changed

+1152
-77
lines changed

5 files changed

+1152
-77
lines changed

README.md

+7-3
Original file line numberDiff line numberDiff line change
@@ -60,11 +60,15 @@ One more thing to the training of custom dataset training, you can set "use_poin
6060
tensorboard logs are exported under "./runs" directory relative to the run script.
6161

6262
## Evaluation
63+
Put your best trained checkpoint under "./src/checkpoints/" directory,
6364

64-
In same train_eval_egnn.py, set the "mode" variable in the main function to "test" mode, then run it to load the test data for evaluation.
65+
For online test mode, In same "src folder" in "eval_egnn_metrics.py", set the "mode" variable in the main function to "test" mode (default is "test"), then run it to load the test split data for evaluation.
6566

66-
For metric results generation, use the `evaluation.py` script located in the `tools` folder to load the saved results of model output in inference model and compared the results under a specified directorh against another directory of gt poses:
67-
$python tools/evaluation_metrics.py
67+
$python src/eval_egnn_metrics.py
68+
69+
Find the average results saved in "output/evaluation_results.txt"
70+
71+
Optionally, For metric results offline generation, use the `evaluation_metrics.py` script located in the `tools` folder to read the saved results saved in txt or pkl of model output in test mode, and compared the results under a specified directorh against another directory of gt poses.
6872

6973
## Citation
7074

datasets/ThreeDMatch.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -443,7 +443,7 @@ def __init__(self,
443443
self.test_file_list = [line.strip() for line in f.readlines()]
444444

445445
def __getitem__(self, index):
446-
file_name = self.file_list[index]
446+
file_name = self.test_file_list[index]
447447

448448
# Load data from .pkl file
449449
with open(os.path.join(self.root, 'train_3dmatch', file_name), 'rb') as f:

0 commit comments

Comments
 (0)