From 8fb0a5eec386868e47d28d432d612b9fa76b067a Mon Sep 17 00:00:00 2001 From: Ethan Weber Date: Tue, 4 Oct 2022 10:36:12 -0700 Subject: [PATCH] adding more datasets (#645) * adding more data * fix name * add more captures --- scripts/downloads/download_data.py | 24 +++++++++++++++++++++--- 1 file changed, 21 insertions(+), 3 deletions(-) diff --git a/scripts/downloads/download_data.py b/scripts/downloads/download_data.py index dc091ceba..2577e8683 100755 --- a/scripts/downloads/download_data.py +++ b/scripts/downloads/download_data.py @@ -48,11 +48,29 @@ def download_friends(save_dir: Path): os.remove(download_path) +def grab_file_id(zip_url: str) -> str: + """Get the file id from the google drive zip url.""" + s = zip_url.split("/d/")[1] + return s.split("/")[0] + + # https://drive.google.com/drive/folders/1Wh66z3qQTZ8o2MwPPwYOrdwQtXUEQFyq?usp=sharing nerfstudio_file_ids = { - "dozer": "1-OR5F_V5S4s-yzxohbwTylaXjzYLu8ZR", - "sf_street": "1DbLyptL6my2QprEVtYuW2uzgp9JAK5Wz", - "poster": "1dmjWGXlJnUxwosN6MVooCDQe970PkD-1", + "dozer": grab_file_id("https://drive.google.com/file/d/1-OR5F_V5S4s-yzxohbwTylaXjzYLu8ZR/view?usp=sharing"), + "sf_street": grab_file_id("https://drive.google.com/file/d/1DbLyptL6my2QprEVtYuW2uzgp9JAK5Wz/view?usp=sharing"), + "poster": grab_file_id("https://drive.google.com/file/d/1dmjWGXlJnUxwosN6MVooCDQe970PkD-1/view?usp=sharing"), + "bunny": grab_file_id("https://drive.google.com/file/d/1oVytSgd1gQclF0CybJ6JdXDmaqKOx7Wg/view?usp=sharing"), + "bww_entrance": grab_file_id("https://drive.google.com/file/d/1ylkRHtfB3n3IRLf2wplpfxzPTq7nES9I/view?usp=sharing"), + "bww_tree": grab_file_id("https://drive.google.com/file/d/1N_OQejT0MblK1UP05KORxy25SXjKqZN1/view?usp=sharing"), + "campanelle": grab_file_id("https://drive.google.com/file/d/13aOfGJRRH05pOOk9ikYGTwqFc2L1xskU/view?usp=sharing"), + "japanese_maple": grab_file_id( + "https://drive.google.com/file/d/1ytCnaAEqm-fIziuQXbBp5yVbZ_cmkHlj/view?usp=sharing" + ), + "kushikatsu": grab_file_id("https://drive.google.com/file/d/1mTNbDW1EyX_fi_ffeP-6Q_0STxr3WfU6/view?usp=sharing"), + "library": grab_file_id("https://drive.google.com/file/d/1Hjbh_-BuaWETQExn2x2qGD74UwrFugHx/view?usp=sharing"), + "redwoods2": grab_file_id("https://drive.google.com/file/d/1rg-4NoXT8p6vkmbWxMOY6PSG4j3rfcJ8/view?usp=sharing"), + "storefront": grab_file_id("https://drive.google.com/file/d/16b792AguPZWDA_YC4igKCwXJqW0Tb21o/view?usp=sharing"), + "vegetation": grab_file_id("https://drive.google.com/file/d/1wBhLQ2odycrtU39y2akVurXEAt9SsVI3/view?usp=sharing"), }