update 3dphotography.py

3dphotography.py

@@ -1,11 +1,13 @@
-# from DepthEstimation.inference import main
-# main()
-# print('Depth Estimation Done!')
-
 import os
 import sys
 from Inpainting import main
-print('impainting started')
-for i in os.listdir(sys.argv[1]):
-  print(f'processing {i}')
-  main.inpaint(i)
+
+print("impainting started")
+
+try:
+    for i in os.listdir(sys.argv[1]):
+        print(f"processing {i}")
+        main.inpaint(i)
+except:
+    print(f"processing {sys.argv[1]}")
+    main.inpaint(sys.argv[1])

Inpainting/main.py

@@ -135,6 +135,8 @@ def inpaint(file_name):
 
         mean_loc_depth = depth[depth.shape[0] // 2, depth.shape[1] // 2]
 
+        starty = time.time()
+
         if not (config["load_ply"] is True and os.path.exists(mesh_fi)):
             vis_photos, vis_depths = sparse_bilateral_filtering(
                 depth.copy(),
@@ -185,7 +187,7 @@ def inpaint(file_name):
                 f"Writing depth ply (and basically doing everything) at {time.time()}"
             )
             # do some mesh work
-            starty = time.time()
+            # starty = time.time()
             rt_info = write_ply(
                 image,
                 depth,