Commit 915fb560 authored by Mohammadreza's avatar Mohammadreza

cr: bug fix

solve the bug of memory leakage after predict
parent b6c36b42
......@@ -43,8 +43,8 @@ file_list = ['./iec_files/1615632656_EyeOpen_1_Other.iec',
'./iec_files/1624281967_EyeOpen_1_Other_0.iec'
]
file_list = [
'./test_iec/1624793690_EyeOpen_20_afshari_0.iec',
# './test_iec/1625637362_EyeOpen_5_maryam_miralmasi_3.iec'
# './test_iec/1624793690_EyeOpen_20_afshari_0.iec',
'./test_iec/1625637362_EyeOpen_5_maryam_miralmasi_3.iec'
# './test_iec/1625758306_EyeOpen_7_asghar_Azimi_3.iec'
]
......@@ -194,7 +194,8 @@ for k in range(0, iec_data.shape[1] - 512, 512):
b = np.max(new_data)
samples = (new_data - a) / (b - a)
samples = samples.reshape((1, new_data.shape[0], 1))
predicted = conv_model.predict(samples).squeeze()
# predicted = conv_model.predict(samples).squeeze()
predicted = conv_model.predict_on_batch(samples).squeeze()
predicted = predicted * (b - a) + a
lst.append([i, j, j + 100, predicted])
else:
......
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