From 7869dc507c7061c82ce08e86498521156e64abec Mon Sep 17 00:00:00 2001 From: cdr2015 Date: Sat, 15 Jan 2022 19:17:09 +0100 Subject: [PATCH] Update extract_official_train_test_set_from_mat.py had to change line 87 to work for me --- utils/extract_official_train_test_set_from_mat.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/extract_official_train_test_set_from_mat.py b/utils/extract_official_train_test_set_from_mat.py index 8bfd257..902d242 100644 --- a/utils/extract_official_train_test_set_from_mat.py +++ b/utils/extract_official_train_test_set_from_mat.py @@ -84,11 +84,11 @@ def convert_image(i, scene, depth_raw, image): print("reading", sys.argv[1]) images = h5_file['images'] - scenes = [u''.join(chr(c) for c in h5_file[obj_ref]) for obj_ref in h5_file['sceneTypes'][0]] + scenes = [u''.join(chr(c[0]) for c in np.array(h5_file[obj_ref])) for obj_ref in h5_file['sceneTypes'][0]] print("processing images") for i, image in enumerate(images): print("image", i + 1, "/", len(images)) convert_image(i, scenes[i], depth_raw[i, :, :].T, image.T) - print("Finished") \ No newline at end of file + print("Finished")