Merge pull request #5413 from Ekhao:wake_vision

PiperOrigin-RevId: 662828687

Author: The TensorFlow Datasets Authors

setup.py

@@ -197,6 +197,7 @@
     'smartwatch_gestures': ['pandas'],
     'svhn': ['scipy'],
     'the300w_lp': ['scipy'],
+    'wake_vision': ['pandas'],
     'wider_face': ['Pillow'],
     'wiki_dialog': ['apache-beam'],
     'wikipedia': ['apache-beam', 'mwparserfromhell', 'mwxml'],

tensorflow_datasets/datasets/wake_vision/CITATIONS.bib

@@ -0,0 +1,6 @@
+@article{banbury2024wake,
+  title={Wake Vision: A Large-scale, Diverse Dataset and Benchmark Suite for TinyML Person Detection},
+  author={Banbury, Colby and Njor, Emil and Stewart, Matthew and Warden, Pete and Kudlur, Manjunath and Jeffries, Nat and Fafoutis, Xenofon and Reddi, Vijay Janapa},
+  journal={arXiv preprint arXiv:2405.00892},
+  year={2024}
+}

tensorflow_datasets/datasets/wake_vision/README.md

@@ -0,0 +1,12 @@
+Wake Vision is a large, high-quality dataset featuring over 6 million images,
+significantly exceeding the scale and diversity of current tinyML datasets
+(100x). This dataset includes images with annotations of whether each image
+contains a person. Additionally, it incorporates a comprehensive fine-grained
+benchmark to assess fairness and robustness, covering perceived gender,
+perceived age, subject distance, lighting conditions, and depictions. The Wake
+Vision labels are derived from Open Image's annotations which are licensed by
+Google LLC under CC BY 4.0 license. The images are listed as having a CC BY 2.0
+license. Note from Open Images: "while we tried to identify images that are
+licensed under a Creative Commons Attribution license, we make no
+representations or warranties regarding the license status of each image and you
+should verify the license for each image yourself."

tensorflow_datasets/datasets/wake_vision/TAGS.txt

@@ -0,0 +1,7 @@
+content.data-type.categorical
+content.data-type.image
+ml.fairness.age
+ml.fairness.facial-attributes
+ml.fairness.gender
+ml.task.fine-grained-image-classification
+ml.task.image-classification

tensorflow_datasets/datasets/wake_vision/__init__.py

@@ -0,0 +1,15 @@
+# coding=utf-8
+# Copyright 2024 The TensorFlow Datasets Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+