search

tensorflow / tpu

Reference models and tools for Cloud TPUs.
https://cloud.google.com/tpu/
Apache License 2.0
5.2k stars 1.77k forks source link

ViLD: issue in tfrecords creation #988

Open xwjabc opened 1 year ago

xwjabc commented 1 year ago

In the preprocessing step for ViLD, a series of tfrecord files should be created. However, there is an issue in the file path of the tfrecords to create. In create_lvis_tf_record.py: https://github.com/tensorflow/tpu/blob/01b67067be64987e9519b35342ba91d3179daeed/models/official/detection/projects/vild/preprocessing/create_lvis_tf_record.py#L215 and https://github.com/tensorflow/tpu/blob/01b67067be64987e9519b35342ba91d3179daeed/models/official/detection/projects/vild/preprocessing/create_lvis_tf_record.py#L266 add FLAGS.dest_dir twice to the path, which incurs issues.

adakan4 commented 1 year ago

I also encountered this error, my solution was to replace osp.join(FLAGS.dest_dir, record_filename)) on line 215 with record_filename).