Bucketed Scene Flow Evaluation
Project description
Bucketed Scene Flow Evaluation
This repo provides the official implementation of Bucket Normalized EPE, as described in our paper I Can't Believe It's Not Scene Flow!
This repo provides:
- A speed and class aware evaluation protocol called Bucket Normalized EPE. See our paper for more details.
- A standardized interface for working with Scene Flow datasets.
- Evaulation infrastructure for the Argoverse 2 2024 Scene Flow Challenge.
Currently supported datasets:
- Argoverse 2 (Human Labeled and NSFP Pseudolabeled)
- Waymo Open (LiDAR only)
- NuScenes (LiDAR only, beta)
If you use this repository as part of a publication, please cite:
@inproceedings{khatri2024trackflow,
author = {Khatri, Ishan and Vedder, Kyle and Peri, Neehar and Ramanan, Deva and Hays, James},
title = {{I Can't Believe It's Not Scene Flow!}},
journal = {European Conference on Computer Vision (ECCV)},
year = {2024},
pdf = {https://arxiv.org/abs/2403.04739},
website={http://vedder.io/trackflow.html},
}
Installation
pip install bucketed-scene-flow-eval
Setup
Follow our Getting Started for setup instructions.
Demo
We provide a demo script which shows off the various features of the API.
Argoverse 2:
To render the lidar and multiple camera views of an Argoverse 2 sequence in 3D, run:
python scripts/demo_3d.py --dataset Argoverse2CausalSceneFlow --root_dir /efs/argoverse2/val/ --with_rgb --sequence_length 4
To render RGB frames with lidar imposed on top, run:
python scripts/demo_rgb.py --dataset Argoverse2SceneFlow --mode project_lidar --reduction_factor 16 --root_dir /efs/argoverse2/val --sequence_length 150 --save_dir /efs/av2_camera_render/
To render the flow field of an Argoverse 2 sequence, run:
python scripts/demo_rgb.py --dataset Argoverse2SceneFlow --mode project_flow --reduction_factor 16 --root_dir /efs/argoverse2/val --sequence_length 150 --save_dir /efs/av2_camera_render/ --flow_dir <path to method flow output>
Waymo Open:
python scripts/demo.py --dataset WaymoOpenSceneFlow --root_dir /efs/waymo_open_processed_flow/validation/
Evaluating AV2 flow submissions
To evaluate an AV2 Scene Flow challenge entry named ./submission_val.zip
against validation dataset masks /efs/argoverse2/val_official_masks.zip
, run
python scripts/av2_eval.py /efs/argoverse2/val /efs/argoverse2/val_official_masks.zip ./submission_val.zip
Documentation
See docs/
for more documentation .
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for bucketed_scene_flow_eval-2.0.18.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 394e9906be537ad875987a9bba5166e75c793a03d17dfee19a5252b5b4cbbff2 |
|
MD5 | aded42929ba952839638ab07f085c95e |
|
BLAKE2b-256 | 2d4e5265c8390793dfea31a6c3368e733a770ed6583b1368562752cf829cdd94 |
Hashes for bucketed_scene_flow_eval-2.0.18-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2b666ec17e702357a6c8c73d2380a1c05f69f639932833e7618b3b3ba5f7b9c3 |
|
MD5 | a8764feed895a8d60702c1bce11e4492 |
|
BLAKE2b-256 | 8a1d27a641402dea2cf3b7828183c9853b85f0a5f2de4ab78ae60defb7e5d2e3 |