search

monocongo / openimages

Tools for downloading images and annotations from Google's OpenImages dataset.
MIT License
47 stars 14 forks source link

API cannot download pictures with spaces in class name #18

Closed GrantPerkins closed 3 years ago

GrantPerkins commented 3 years ago

If a class name has a space, this API will fatally error.

download_dataset(dest_dir="./data", csv_dir="./data", class_labels=["remote_control"],
  File "C:\Python39\lib\site-packages\openimages\download.py", line 133, in download_dataset
    label_codes = _class_label_codes(class_labels, csv_dir)
  File "C:\Python39\lib\site-packages\openimages\download.py", line 94, in _class_label_codes
    df_classes.loc[df_classes[1] == class_label].values[0][0]
IndexError: index 0 is out of bounds for axis 0 with size 0
GrantPerkins commented 3 years ago

Code to reproduce:

from openimages.download import download_dataset
download_dataset(dest_dir="./data", csv_dir="./data", class_labels=["remote_control"],
                     annotation_format="pascal", exclusions_path=None, limit=5)