perf: faster initialize env.

app.py

@@ -87,20 +87,19 @@ def setup_runtime_env():
     logging.info("CUDA is available: %s" % torch.cuda.is_available())
     logging.info("CUDA Device Capability: %s" % (torch.cuda.get_device_capability(),))
 
-    # Install Pre-compiled CUDA extensions (Not working)
+    # Install Pre-compiled CUDA extensions
     # Ref: https://huggingface.co/spaces/zero-gpu-explorers/README/discussions/110
-    #
-    # ext_dir = os.path.join(os.path.dirname(__file__), "wheels")
-    # for e in os.listdir(ext_dir):
-    #     logging.info("Installing Extensions from %s" % e)
-    #     subprocess.call(
-    #         ["pip", "install", os.path.join(ext_dir, e)], stderr=subprocess.STDOUT
-    #     )
-    # Compile CUDA extensions
-    ext_dir = os.path.join(os.path.dirname(__file__), "gaussiancity", "extensions")
+    ext_dir = os.path.join(os.path.dirname(__file__), "wheels")
     for e in os.listdir(ext_dir):
-        if os.path.isdir(os.path.join(ext_dir, e)):
-            subprocess.call(["pip", "install", "."], cwd=os.path.join(ext_dir, e))
+        logging.info("Installing Extensions from %s" % e)
+        subprocess.call(
+            ["pip", "install", os.path.join(ext_dir, e)], stderr=subprocess.STDOUT
+        )
+    # Compile CUDA extensions
+    # ext_dir = os.path.join(os.path.dirname(__file__), "gaussiancity", "extensions")
+    # for e in os.listdir(ext_dir):
+    #     if os.path.isdir(os.path.join(ext_dir, e)):
+    #         subprocess.call(["pip", "install", "."], cwd=os.path.join(ext_dir, e))
 
     logging.info("Installed Python Packages: %s" % _get_output(["pip", "list"]))
 
@@ -218,14 +217,14 @@ if __name__ == "__main__":
         format="[%(levelname)s] %(asctime)s %(message)s", level=logging.INFO
     )
     logging.info("Environment Variables: %s" % os.environ)
-    if _get_output(["nvcc", "--version"]) is None:
-        logging.info("Installing CUDA toolkit...")
-        install_cuda_toolkit()
-    else:
-        logging.info("Detected CUDA: %s" % _get_output(["nvcc", "--version"]))
+    # if _get_output(["nvcc", "--version"]) is None:
+    #     logging.info("Installing CUDA toolkit...")
+    #     install_cuda_toolkit()
+    # else:
+    #     logging.info("Detected CUDA: %s" % _get_output(["nvcc", "--version"]))
 
     logging.info("Compiling CUDA extensions...")
-    # setup_runtime_env()
+    setup_runtime_env()
 
     logging.info("Downloading pretrained models...")
     fgm = get_models("GaussianCity-Fgnd.pth")

gaussiancity/inference.py

@@ -52,7 +52,6 @@ def get_instance_seg_map(seg_map):
     # Mapping constructions to buildings
     seg_map[seg_map == CLASSES["CONSTRUCTION"]] = CLASSES["BLDG_FACADE"]
     # Use connected components to get building instances
-    import pdb; pdb.set_trace()
     _, labels, _, _ = cv2.connectedComponentsWithStats(
         (seg_map == CLASSES["BLDG_FACADE"]).astype(np.uint8), connectivity=4
     )