Upload luciddreamer.py with huggingface_hub

luciddreamer.py

@@ -231,7 +231,7 @@ class LucidDreamer:
         framelist = []
         depthlist = []
         dmin, dmax = 1e8, -1e8
-        for view in views:
+        for view in progress.tqdm(views, desc='[4/4] Rendering a video'):
             results = render(view, self.gaussians, self.opt, self.background, render_only=True)
             frame, depth = results['render'], results['depth']
             framelist.append(
@@ -244,6 +244,9 @@ class LucidDreamer:
             if dmax_local > dmax:
                 dmax = dmax_local
             depthlist.append(depth)
+
+        progress(1, desc='[4/4] Rendering a video...')
+
         # depthlist = [colorize(depth, vmin=dmin, vmax=dmax) for depth in depthlist]
         depthlist = [colorize(depth) for depth in depthlist]
         if not os.path.exists(videopath):
@@ -256,7 +259,7 @@ class LucidDreamer:
         if not self.scene:
             raise('Build 3D Scene First!')
         
-        for iteration in tqdm(range(1, self.opt.iterations + 1)):
+        for iteration in progress.tqdm(range(1, self.opt.iterations + 1), desc='[3/4] Baking Gaussians'):
             self.gaussians.update_learning_rate(iteration)
 
             # Every 1000 its we increase the levels of SH up to a maximum degree
@@ -349,10 +352,10 @@ class LucidDreamer:
 
         pts_coord_world, pts_colors = new_pts_coord_world2.copy(), new_pts_colors2.copy()
 
-        progress(0, desc='Dreaming...')
+        progress(0, desc='[1/4] Dreaming...')
         # time.sleep(0.5)
 
-        for i in progress.tqdm(range(1, len(render_poses)), desc='Dreaming'):
+        for i in progress.tqdm(range(1, len(render_poses)), desc='[1/4] Dreaming'):
             R, T = render_poses[i,:3,:3], render_poses[i,:3,3:4]
 
             ### Transform world to pixel
@@ -480,10 +483,10 @@ class LucidDreamer:
         # render_poses = get_pcdGenPoses(pcdgenpath)
         internel_render_poses = get_pcdGenPoses('hemisphere', {'center_depth': center_depth})
 
-        progress(0, desc='Aligning...')
+        progress(0, desc='[2/4] Aligning...')
         # time.sleep(0.5)
 
-        for i in progress.tqdm(range(len(render_poses)), desc='Aligning'):
+        for i in progress.tqdm(range(len(render_poses)), desc='[2/4] Aligning'):
             for j in range(len(internel_render_poses)):
                 idx = i * len(internel_render_poses) + j
                 print(f'{idx+1} / {len(render_poses)*len(internel_render_poses)}')
@@ -539,5 +542,5 @@ class LucidDreamer:
                     'transform_matrix': Pc2w.tolist(),
                 })
 
-        progress(1, desc='Baking Gaussians...')
+        progress(1, desc='[3/4] Baking Gaussians...')
         return traindata