tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-10 18:17:31 +08:00
Code

Check if curl is installed (#1337)

Browse Source 
* check if curl is installed

* lint
This commit is contained in:
Matthew Tancik 2023-02-02 10:11:41 -08:00 committed by GitHub
parent d547279a92
commit ef9e00e614
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nerfstudio/utils/install_checks.py
View File

@ -39,3 +39,11 @@ def check_colmap_installed():
CONSOLE.print("[bold red]Could not find COLMAP. Please install COLMAP.")
print("See https://colmap.github.io/install.html for installation instructions.")
sys.exit(1)
def check_curl_installed():
"""Checks if curl is installed."""
curl_path = shutil.which("curl")
if curl_path is None:
CONSOLE.print("[bold red]Could not find [yellow]curl[red], Please install [yellow]curl")
sys.exit(1)

3
scripts/downloads/download_data.py
View File

@ -15,6 +15,7 @@ from rich.console import Console
from typing_extensions import Annotated
from nerfstudio.configs.base_config import PrintableConfig
from nerfstudio.utils import install_checks
from nerfstudio.utils.scripts import run_command
CONSOLE = Console(width=120)
@ -212,6 +213,7 @@ class DNerfDownload(DatasetDownload):
"""Download the D-NeRF dataset (https://github.com/albertpumarola/D-NeRF)."""
# TODO: give this code the same structure as download_nerfstudio
install_checks.check_curl_installed()
final_path = save_dir / Path("dnerf")
if os.path.exists(final_path):
shutil.rmtree(str(final_path))
@ -253,6 +255,7 @@ class PhototourismDownload(DatasetDownload):
def download(self, save_dir: Path):
"""Download a PhotoTourism dataset: https://www.cs.ubc.ca/~kmyi/imw2020/data.html"""
install_checks.check_curl_installed()
if self.capture_name == "all":
for capture_name in phototourism_downloads:
if capture_name != "all":