old_workbook = xlrd.open_workbook("testdata.xlsx")
old_sheet = old_workbook.sheet_by_index(0)
old_n_row = old_sheet.nrows
old_n_col = 2
# 建新的文件
new_workbook = xlsxwriter.Workbook("testdata.xlsx")
new_sheet = new_workbook.add_worksheet()
# 复制当前已有的数据
for i in range(old_n_row):
for j in range(old_n_col):
new_sheet.write(i, j, old_sheet.row_values(i)[j])
model_path = r'Saved model/loss=1.562.tar'
device = device("cuda:0" if cuda.is_available() else "cpu")
model = CNN.FFTCNN()
model.load_state_dict(load(model_path, map_location='cpu'))
model = model.to(device)
model.eval()
acc_list = [0] * 4
testkey = input("Please input test key type:\n")
FILENAME = 'pred'
##### 接受录音、截取处理成fft
wavfile = Record(5, FILENAME)
# wavfile = r'recordings\b1'
since = time()
fft, _ = Get_fft(wavfile + '.wav')
if len(fft.shape) == 1: # 只截取到一个按键音
fft = fft.reshape(1, -1)
