This repository provides tools and scripts for visualizing and reconstructing the EPIC FIELDS dataset.
This visualisation code is associated with the released EPIC FIELDS dataset. Further details on the dataset and associated preprint are available at: https://epic-kitchens.github.io/epic-fields
- The
cameraparameters use the COLMAP format, which is the same as the OpenCV format. - The
imagesstores the world-to-camera transformation, represented by quaternion and translation.- Note: for NeRF usage this needs to be converted to camera-to-world transformation and possibly changing (+x, +y, +z) to (+x, -y, -z)
- The
pointsis part of COLMAP output. It's kept here for visualisation purpose and potentially for computing thenear/farbounds in NeRF input.
{
"camera": {
"id": 1, "model": "OPENCV", "width": 456, "height": 256,
"params": [fx, fy, cx, cy, k1, k2, p1, p2]
},
"images": {
frame_name: [qw, qx, qy, qz, tx, ty, tz],
...
},
"points": [
[x, y, z, r, g, b],
...
]
}
example data can be found in `example_data/P28_101.json`
This script requires Open3D. This script is tested with Open3D==0.16.1.
python tools/visualise_data_open3d.py --json-data example_data/P28_101.jsonPS: Press 'h' to see the Open3D help message.
Click to see the example output
python tools/project_3d_line.py \
--json-data example_data/P28_101.json \
--line-data example_data/P28_101_line.json \
--frames-root example_data/P28_101/Click to see the example output
To draw a 3D line, one option is to download the COLMAP format data and use COLMAP GUI to click on points.
This section contains the pipeline for the dataset introduced in our paper, "EPIC Fields: Marrying 3D Geometry and Video Understanding." We aim to bridge the domains of 3D geometry and video understanding, leading to innovative advancements in both areas.
This section outlines the procedure to achieve the EPIC-KITCHENS reconstructions using our methodology.
To efficiently process and reconstruct the frames, it's recommended to install COLMAP with CUDA support, which accelerates the reconstruction process using NVIDIA GPUs.
You can download and install COLMAP from their official website. For detailed installation instructions, especially on how to enable CUDA support, refer to the COLMAP installation guide.
To proceed with the subsequent steps, you'll need to clone the current repository. Run the following commands:
git clone https://github.com/epic-kitchens/epic-fields-code.git
cd epic-fields-codeCOLMAP utilizes vocabulary trees for efficient image matching. Create a directory called vocab_bins and download the required Vocabulary Trees into this directory:
mkdir vocab_bins
cd vocab_bins
wget https://demuc.de/colmap/vocab_tree_flickr100K_words32K.bin
cd ..The pycolmap package will be used to gather statistics from the model later on. Install it using pip (assuming that you've created an environment):
pip install pycolmapTo utilize the EPIC Fields pipeline, the first step is to acquire the necessary video frames. We're particularly interested in the RGB frames from EPIC-KITCHENS. You can download the entire collection from EPIC-KITCHENS.
For demonstration purposes, we'll guide you through downloading the P15_12 video RGB frames.
Execute the following shell commands to download and extract the RGB frames:
# Download the tarball
wget https://data.bris.ac.uk/datasets/3h91syskeag572hl6tvuovwv4d/frames_rgb_flow/rgb/train/P15/P15_12.tar
# Create the desired directory structure
mkdir -p P15/P15_12
# Extract the frames into the specified directory
tar -xf P15_12.tar -C P15/P15_12This will place all the .jpg frames inside the P15/P15_12 directory.
After downloading and extracting, your directory structure should look like this (which is EPIC-KITCHENS format :
/root-directory/
│
└───PXX/
│
└───PXX_YY(Y)/
│ frame_000001.jpg
│ frame_000002.jpg
│ ...
For our P15_12 example, this would be:
/root-directory/
│
└───P15/
│
└───P15_12/
│ frame_000001.jpg
│ frame_000002.jpg
│ ...
This structure ensures a consistent format for the pipeline to process the frames effectively.
Update the input_videos.txt file in the repository to list the video identifiers you wish to process. In our demo example, we put P15_12 in the file. If you have multiple files, please ensure each video identifier is on a separate line.
Execute the select_sparse_frames.py script to perform homography-based sampling of the frames.
--input_videos: Path to the file containing a list of videos to be processed. Default:input_videos.txt--epic_kithens_root: Directory path to the EPIC-KITCHENS images. Default:.--sampled_images_path: Directory where the sampled image files will be stored. Default:sampled_frames--homography_overlap: Threshold for the homography to sample new frames. A higher value will sample more images. Default:0.9--max_concurrent: Maximum number of concurrent processes. Default:8
python3 select_sparse_frames.py --input_videos input_videos.txt --epic_kithens_root path_to_epic_images --sampled_images_path path_for_sampled_framesFor the demo, using the P15_12 video you've downloaded into the current directory, run:
python3 select_sparse_frames.py --input_videos input_videos.txt --epic_kithens_root . --sampled_images_path sampled_frames --homography_overlap 0.9 --max_concurrent 8Execute the reconstruct_sparse.py script to perform sparse reconstruction using COLMAP.
--input_videos: Path to the file containing a list of videos to be processed. Default:input_videos.txt--sparse_reconstuctions_root: Path to store the sparsely reconstructed models. Default:colmap_models/sparse--epic_kithens_root: Directory path to the EPIC-KITCHENS images. Default:.--logs_path: Path where the log files will be stored. Default:logs/sparse/out_logs_terminal--summary_path: Path where the summary files will be stored. Default:logs/sparse/out_summary--sampled_images_path: Directory where the sampled image files are located. Default:sampled_frames--gpu_index: Index of the GPU to be used. Default:0
python3 reconstruct_sparse.py --input_videos input_videos.txt --sparse_reconstuctions_root colmap_models/sparse --epic_kithens_root path_to_epic_images --logs_path logs/sparse/out_logs_terminal --summary_path logs/sparse/out_summary --sampled_images_path path_for_sampled_frames --gpu_index 0For the demo, using the P15_12 video and the sampled frames in the current directory, run:
python3 reconstruct_sparse.py --input_videos input_videos.txt --sparse_reconstuctions_root colmap_models/sparse --epic_kithens_root . --logs_path logs/sparse/out_logs_terminal --summary_path logs/sparse/out_summary --sampled_images_path sampled_frames --gpu_index 0After running the sparse reconstruction demo, you'll notice the following directory hierarchy:
logs/
│
└───sparse/
│
├───out_logs_terminal/
│ │ P15_12_<PROCESS_ID>_reconstruct_sparse.out
│ │ ...
│
└───out_summary/
│ P15_12.out
│ ...
The sparsely reconstructed model for our demo video P15_12 will be found in: colmap_models/sparse/P15_12
The "logs" directory provides insights into the sparse reconstruction process:
-
COLMAP Execution Logs (out_logs_terminal): These logs capture details from the COLMAP execution and can be helpful for debugging. For our demo video P15_12, the respective log file would be named something like:
logs/sparse/out_logs_terminal/P15_12_<PROCESS_ID>_reconstruct_sparse.out -
Sparse Model Summary (out_summary): This directory contains a summary of the sparse model's statistics. For our demo video P15_12, the summary file is
logs/sparse/out_summary/P15_12.outBy examining the P15_12.out file, you can gain insights into how well the reconstruction process performed for that specific video and the excution time.
For this step, you'll use the register_dense.py script. This script registers all the frames with the sparse model, preparing them for a dense reconstruction.
--input_videos: Path to the file containing a list of videos to be processed. Default:input_videos.txt--sparse_reconstuctions_root: Directory path to the sparsely reconstructed models. Default:colmap_models/sparse--dense_reconstuctions_root: Directory path to the densely registered models. Default:colmap_models/dense--epic_kithens_root: Directory path to the EPIC-KITCHENS images. Default:.--logs_path: Directory where the log files of the dense registration will be stored. Default:logs/dense/out_logs_terminal--summary_path: Directory where the summary files of the dense registration will be stored. Default:logs/dense/out_summary--gpu_index: Index of the GPU to use. Default:0
To demonstrate the registration process using the register_dense.py script, let's use the sample video P15_12 as an example.
python3 register_dense.py --input_videos input_videos.txt --sparse_reconstuctions_root colmap_models/sparse --dense_reconstuctions_root colmap_models/dense --epic_kithens_root . --logs_path logs/dense/out_logs_terminal --summary_path logs/dense/out_summary --gpu_index 0Assuming input_videos.txt contains the entry for P15_12, the above command will register all frames from the P15_12 video with the sparse model stored under colmap_models/sparse, the new registered model will be saved under colmap_models/dense. The logs and summary for this registration process will be saved under the logs/dense/out_logs_terminal and logs/dense/out_summary directories, respectively.
After executing the command, you can check the log files and summary for insights and statistics on the registration process for the P15_12 video.
Here we provide another demo script demo/demo.py
that works on a video directly, summarising all above steps into one file.
python demo/demo.py video.mp4
Please refer to demo_ego4d.md for details.
Code prepared by Zhifan Zhu, Ahmad Darkhalil and Vadim Tschernezki.
If you find this work useful please cite our paper:
@article{EPICFIELDS2023,
title={{EPIC-FIELDS}: {M}arrying {3D} {G}eometry and {V}ideo {U}nderstanding},
author={Tschernezki, Vadim and Darkhalil, Ahmad and Zhu, Zhifan and Fouhey, David and Larina, Iro and Larlus, Diane and Damen, Dima and Vedaldi, Andrea},
booktitle = {ArXiv},
year = {2023}
}
Also cite the EPIC-KITCHENS-100 paper where the videos originate:
@ARTICLE{Damen2022RESCALING,
title={Rescaling Egocentric Vision: Collection, Pipeline and Challenges for EPIC-KITCHENS-100},
author={Damen, Dima and Doughty, Hazel and Farinella, Giovanni Maria and and Furnari, Antonino
and Ma, Jian and Kazakos, Evangelos and Moltisanti, Davide and Munro, Jonathan
and Perrett, Toby and Price, Will and Wray, Michael},
journal = {International Journal of Computer Vision (IJCV)},
year = {2022},
volume = {130},
pages = {33–55},
Url = {https://doi.org/10.1007/s11263-021-01531-2}
}
For more information on the project and related research, please visit the EPIC-Kitchens' EPIC Fields page.
All files in this dataset are copyright by us and published under the Creative Commons Attribution-NonCommerial 4.0 International License, found here. This means that you must give appropriate credit, provide a link to the license, and indicate if changes were made. You may do so in any reasonable manner, but not in any way that suggests the licensor endorses you or your use. You may not use the material for commercial purposes.
For general enquiries regarding this work or related projects, feel free to email us at uob-epic-kitchens@bristol.ac.uk.