Commit 021828cc authored by Meghna Natraj's avatar Meghna Natraj Committed by Sergio Guadarrama
Browse files

Create visualwakewords dataset using slim scripts instead of custom scripts (#7796)

* Internal changes to tf.contrib symbols

PiperOrigin-RevId: 277940048

* Create visualwakewords dataset using slim scripts instead of custom scripts.

PiperOrigin-RevId: 278917344

* update for python3.0 compatibility
parent 81d031d0
......@@ -201,7 +201,7 @@ def create_tf_record_for_visualwakewords_dataset(annotations_file, image_dir,
groundtruth_data = json.load(fid)
images = groundtruth_data['images']
annotations_index = groundtruth_data['annotations']
annotations_index = {int(k): v for k, v in annotations_index.iteritems()}
annotations_index = {int(k): v for k, v in annotations_index.items()}
# convert 'unicode' key to 'int' key after we parse the json file
for idx, image in enumerate(images):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment