| |
| from __future__ import annotations |
|
|
| import argparse |
| from pathlib import Path |
| from urllib.request import urlretrieve |
|
|
| from tqdm import tqdm |
|
|
| BASE_URL = "https://fenix.ur.edu.pl/~mkepski/ds/data" |
|
|
|
|
| def download(url: str, destination: Path) -> None: |
| destination.parent.mkdir(parents=True, exist_ok=True) |
| if destination.exists() and destination.stat().st_size > 0: |
| return |
| temporary = destination.with_suffix(destination.suffix + ".part") |
| urlretrieve(url, temporary) |
| temporary.replace(destination) |
|
|
|
|
| def main() -> None: |
| parser = argparse.ArgumentParser(description="Download official URFD RGB videos") |
| parser.add_argument("--output", type=Path, default=Path("data/videos")) |
| args = parser.parse_args() |
|
|
| items: list[tuple[str, Path]] = [] |
| for index in range(1, 31): |
| name = f"fall-{index:02d}-cam0.mp4" |
| items.append((f"{BASE_URL}/{name}", args.output / "fall" / f"fall-{index:02d}" / name)) |
| for index in range(1, 41): |
| name = f"adl-{index:02d}-cam0.mp4" |
| items.append((f"{BASE_URL}/{name}", args.output / "normal" / f"adl-{index:02d}" / name)) |
| items.extend( |
| [ |
| ( |
| f"{BASE_URL}/urfall-cam0-falls.csv", |
| args.output / "annotations" / "urfall-cam0-falls.csv", |
| ), |
| ( |
| f"{BASE_URL}/urfall-cam0-adls.csv", |
| args.output / "annotations" / "urfall-cam0-adls.csv", |
| ), |
| ] |
| ) |
| for url, destination in tqdm(items, desc="Downloading URFD"): |
| download(url, destination) |
| print(f"Downloaded/verified {len(items)} files in {args.output}") |
| print("License: CC BY-NC-SA 4.0, academic non-commercial use.") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
|
|
