Skip to content

Instantly share code, notes, and snippets.

@chricke
Last active Mar 28, 2019
Embed
What would you like to do?
create TFRecords
lumi dataset transform --type csv --data-dir data/ --output-dir tfdata/ --split train --split val --only-classes=table
@goodhamgupta
Copy link

goodhamgupta commented Jan 16, 2019

Hi @jyoti59n,

There is a problem with the given train.csv and val.csv files. The header rows are missing because of which you get the missing columns error(Check the _check_csv function of the csv_reader in luminoth here).

Add the following to the first line of both the flies:

image_id,xmin,ymin,xmax,ymax,label

It should work after this change.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment