chen-zichen / nutrition5k

Detailed visual + nutritional data for over 5,000 plates of food.

Nutrition5k: A Comprehensive Nutrition Dataset

Paper • Download Data • Dataset Contents • License & Contact

Nutrition5k is a dataset of visual and nutritional data for ~5k realistic plates of food captured from Google cafeterias using a custom scanning rig. We are releasing this dataset alongside our recent CVPR 2021 paper to help promote research in visual nutrition understanding. Please see the paper for more details on the dataset and follow-up experiments.

Key Features

• Scans data for 5,006 plates of food, each containing:
  • 4 rotating side-angle videos
  • Overhead RGB-D images (when available)
  • Fine-grained list of ingredients
  • Per-ingredient mass
  • Total dish mass and calories
  • Fat, protein, and carbohydrate macronutrient masses
• Official train/test split
• Nutrition regression eval scripts

→ [Also, see our related Mobile Food Segmentation model on TensorFlow Hub]

Download Data

All Nutrition5k data can be downloaded directly from our Google Cloud Storage bucket, or from the .tar.gz download link below.

• nutrition5k_dataset.tar.gz (181.4 GB)

From the Cloud Storage bucket directory, you can also browse through the dataset folders and download specific files using the gsutil cp command:

gsutil -m cp -r "gs://nutrition5k_dataset/nutrition5k_dataset/{FILE_OR_DIR_PATH}" .

See here for instructions on installing the gsutil tool.

Examples

Example side-angle and overhead frames, with nutrition labels.

Example of the incremental scanning procedure.

Dataset contents

Side-Angle Videos

Video recordings were captured using 4 separate Raspberry Pi cameras (labeled A-D) at alternating 30 degree and 60 degree viewing angles. The cameras are positioned 90 degrees apart and sweep 90 degrees during video capture so that the dish is captured from all sides. Video files are found in imagery/side_angles/ and are organized by dish id. To extract all 2D image frames from a video, use ffmpeg as shown below:

ffmpeg -i input.mp4 output_%03d.jpeg

The models included in the Nutrition5k paper were trained and evaluated on every fifth frame sampled from each video, and we include two scripts to help with this. scripts/extract_frames.sh walks through each dish_id directory, extracts each frame from all four videos, and outputs the frames to a newly created dish_ID/frames/ directory. scripts/extract_frames_sampled.sh accepts a command line argument to extract only every nth frame from each video.

cp scripts/extract_frames*.sh imagery/side_angles
cd imagery/side_angles
sh extract_frames.sh
sh extract_frames_sampled.sh 5

Overhead RGB-D Images

The imagery/realsense_overhead/ directory contains RGB, raw depth, and colorized depth images organized by dish ID. Raw depth images are encoded as 16-bit integer images with depth units of 10,000 (i.e. 1 meter = 10,000 units). The colorized depth images provide a human-readable visualization of the depth map, with closer objects in blue and further objects in red. All depth values are rounded to a maximum of 0.4m (4,000 depth units), which exceeds the height of our food scanning rig.

Ingredient Metadata

The ingredient metadata CSV (metadata/ingredient_metadata.csv) contains a list of all ingredients covered in the dataset's dishes, their unique IDs, and per-gram nutritional information sourced from the USDA Food and Nutrient Database. Ingredient IDs take the following form: ingr_[ingredient number padded to 10 digits].

Dish Metadata

The dish metadata CSVs (metadata/dish_metadata_cafe1.csv and metadata/dish_metadata_cafe2.csv) contain all nutrition metadata at the dish-level, as well as per-ingredient mass and macronutrients. For each dish ID dish_[10 digit timestamp], there is a CSV entry containing the following fields:

dish_id, total_calories, total_mass, total_fat, total_carb, total_protein, num_ingrs, (ingr_1_id, ingr_1_name, ingr_1_grams, ingr_1_calories, ingr_1_fat, ingr_1_carb, ingr_1_protein, ...)

with the last 8 fields are repeated for every ingredient present in the dish.

Train/Test Splits

We include dish IDs for the training and testing splits used in our experiments in the dish_ids/splits/ directory. All incremental scans that compose a unique plate are held within the same split, to avoid overlap between the train and test splits. See Section 3.6 of our paper for more details on incremental scanning.

Evaluation Script

To help evaluate nutrition prediction methods, we provide scripts/compute_eval_statistics.py. This script can be used to calculate absolute and percentage mean average error from a CSV of per-dish nutrition values. This tool can be used to generate regression results that can be directly compared to those reported in our paper. See the header file for usage instructions.

Dataset Bias Disclaimer

The dataset does not cover all food cuisines, as it was only collected in a few select cafeterias in California, USA. Nutrition5k does not claim to completely solve the food understanding problem, but rather aims to provide a unique level of detailed annotations and depth data to further advance the space.

License & Contact

We release all Nutrition5k data under the Creative Commons V4.0 license. You are free to share and adapt this data for any purpose, even commercially. If you found this dataset useful, please consider citing our CVPR 2021 paper.

@inproceedings{thames2021nutrition5k,
  title={Nutrition5k: Towards Automatic Nutritional Understanding of Generic Food},
  author={Thames, Quin and Karpur, Arjun and Norris, Wade and Xia, Fangting and Panait, Liviu and Weyand, Tobias and Sim, Jack},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={8903--8911},
  year={2021}
}

If you have any questions about the Nutrition5k dataset or paper, please send an email to the authors at [email protected].

Thank you to Ben Goldberger, Caitlin O'Brien, and the Google LAX/PLV Cafeteria teams for their contributions!