search

EscVM / OIDv4_ToolKit

Download and visualize single or multiple classes from the huge Open Images v4 dataset
GNU General Public License v3.0
798 stars 633 forks source link

IndexError: index 0 is out of bounds for axis 0 with size 0 #112

Open lordwildbeast opened 1 year ago

lordwildbeast commented 1 year ago

i met an error while I ran the following commands within the toolkit to create my custom training dataset. python main.py downloader --classes 'Vehicle_registration_plate' --type_csv train --limit 15

do you know how to solve it? thanks

image

ThinkWD commented 1 year ago

This is because the passed in classes cannot find the same value in OID/csv_folder/class-descriptions-boxable.csv

siri-g-265 commented 1 year ago

This issue is faced when the class names are more than one word.

Solution: Make a classes.txt file with all the class names you want to download and send the path python main.py downloader --classes OID/classes.txt --type_csv train --limit 15