xref: /aosp_15_r20/external/rnnoise/training/bin2hdf5.py (revision 1295d6828459cc82c3c29cc5d7d297215250a74b)
1*1295d682SXin Li#!/usr/bin/python
2*1295d682SXin Li
3*1295d682SXin Lifrom __future__ import print_function
4*1295d682SXin Li
5*1295d682SXin Liimport numpy as np
6*1295d682SXin Liimport h5py
7*1295d682SXin Liimport sys
8*1295d682SXin Li
9*1295d682SXin Lidata = np.fromfile(sys.argv[1], dtype='float32');
10*1295d682SXin Lidata = np.reshape(data, (int(sys.argv[2]), int(sys.argv[3])));
11*1295d682SXin Lih5f = h5py.File(sys.argv[4], 'w');
12*1295d682SXin Lih5f.create_dataset('data', data=data)
13*1295d682SXin Lih5f.close()
14