Scripts for making JSON files of the test_init_poses
taeyeopl opened this issue · 1 comments
taeyeopl commented
Can you share the scripts for making JSON files of the test_init_poses?
I checked three methods (Dualpose, NOCS, SPD) of initial poses are shared, and I want to adopt different initial poses.
https://github.com/THU-DA-6D-Pose-Group/CATRE/tree/main/datasets/NOCS/test_init_poses
It will really help to adapt your proposed method to different pose methods with different initial poses.
example,
INIT_POSE_FILES_TEST=("datasets/NOCS/test_init_poses/init_pose_spd_nocs_real.json",),
INIT_POSE_FILES_TEST=("datasets/NOCS/test_init_poses/init_pose_dualposenet_nocs_real.json",),
taeyeopl commented
Sorry, I think I found the example script here.
https://github.com/THU-DA-6D-Pose-Group/CATRE/blob/main/core/catre/tools/prepare_spd_init_results.py