@@ -106,17 +106,16 @@ def get_ground_truth(pose, cur_scan, seq_map, max_range):
106
106
@click .option ('--path' , '-p' , type = str , default = '' , help = 'path to the scan sequence' )
107
107
@click .option ('--voxel_size' , '-v' , type = float , default = 0.05 , help = 'voxel size' )
108
108
@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 )
113
115
114
116
poses = load_poses (os .path .join (PATH_DATA , 'calib.txt' ), os .path .join (PATH_DATA , 'poses.txt' ))
115
117
seq_map = np .load (f'{ PATH_DATA } /map_clean.npy' )
116
118
117
- if not diff :
118
- print (f'Evaluating baseline { path .split ("/" )[- 3 ]} ' )
119
-
120
119
jsd_3d = []
121
120
jsd_bev = []
122
121
0 commit comments