1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
import h5py
from pathlib import Path
import numpy as np
import json
import matplotlib.pyplot as plt
import matplotlib.patches as patches
import argparse
def show(file: h5py.File):
fig = plt.figure()
rgb = np.array(file['colors'])
text = np.array(f["object_data"]).tostring()
print(text)
if len(text) > 0:
object_data = json.loads(text)
for object in object_data:
if 'bounding_box' in object:
bb = object['bounding_box']
rect = patches.Rectangle((bb[0], bb[1]), bb[2], bb[3], linewidth=1, edgecolor='r', facecolor='none')
plt.gca().add_patch(rect)
plt.imshow(rgb)
plt.show()
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--path', type=str, help='Path to a folder containing hdf5 files')
args = parser.parse_args()
folder = Path(args.path)
assert folder.exists()
for file in folder.iterdir():
if file.name.endswith('.hdf5'):
with h5py.File(str(file)) as f:
print(f'Visualizing file {f}')
show(f)
|
