Skip to content

Commit b0f2cc0

Browse files
committed
fix eval script
1 parent 5228cac commit b0f2cc0

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

lidiff/utils/eval_path.py

+6-7
Original file line numberDiff line numberDiff line change
@@ -106,17 +106,16 @@ def get_ground_truth(pose, cur_scan, seq_map, max_range):
106106
@click.option('--path', '-p', type=str, default='', help='path to the scan sequence')
107107
@click.option('--voxel_size', '-v', type=float, default=0.05, help='voxel size')
108108
@click.option('--max_range', '-m', type=float, default=50, help='max range')
109-
@click.option('--ckpt_path', '-c', type=str, help='path to the checkpoint for diffusion pipeline')
110-
@click.option('--diff', '-d', is_flag=True, help='run diffusion pipeline')
111-
def main(path, voxel_size, max_range, ckpt_path, diff):
112-
diff_completion = DiffCompletion(ckpt_path) if diff else None
109+
@click.option('--denoising_steps', '-t', type=int, default=50, help='number of denoising steps')
110+
@click.option('--cond_weight', '-s', type=float, default=6.0, help='conditioning weights')
111+
@click.option('--diff', '-d', type=str, help='run diffusion pipeline')
112+
@click.option('--refine', '-r', type=str, help='path to the checkpoint for refinement net')
113+
def main(path, voxel_size, max_range, denoising_steps, cond_weight, diff, refine):
114+
diff_completion = DiffCompletion(diff, refine, denoising_steps, cond_weight)
113115

114116
poses = load_poses(os.path.join(PATH_DATA, 'calib.txt'), os.path.join(PATH_DATA, 'poses.txt'))
115117
seq_map = np.load(f'{PATH_DATA}/map_clean.npy')
116118

117-
if not diff:
118-
print(f'Evaluating baseline {path.split("/")[-3]}')
119-
120119
jsd_3d = []
121120
jsd_bev = []
122121

0 commit comments

Comments
 (0)