Skip to content

S
Signal Denoise with AutoEncoder
Commit 915fb560 authored by Mohammadreza's avatar Mohammadreza
Browse files
Options

cr: bug fix 

solve the bug of memory leakage after predict
parent b6c36b42
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
test_file_traverse.py
View file @ 915fb560
...@@ -43,8 +43,8 @@ file_list = ['./iec_files/1615632656_EyeOpen_1_Other.iec', ...@@ -43,8 +43,8 @@ file_list = ['./iec_files/1615632656_EyeOpen_1_Other.iec',
'./iec_files/1624281967_EyeOpen_1_Other_0.iec' './iec_files/1624281967_EyeOpen_1_Other_0.iec'
] ]
file_list = [ file_list = [
'./test_iec/1624793690_EyeOpen_20_afshari_0.iec', # './test_iec/1624793690_EyeOpen_20_afshari_0.iec',
# './test_iec/1625637362_EyeOpen_5_maryam_miralmasi_3.iec' './test_iec/1625637362_EyeOpen_5_maryam_miralmasi_3.iec'
# './test_iec/1625758306_EyeOpen_7_asghar_Azimi_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): ...@@ -194,7 +194,8 @@ for k in range(0, iec_data.shape[1] - 512, 512):
b = np.max(new_data) b = np.max(new_data)
samples = (new_data - a) / (b - a) samples = (new_data - a) / (b - a)
samples = samples.reshape((1, new_data.shape[0], 1)) 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 predicted = predicted * (b - a) + a
lst.append([i, j, j + 100, predicted]) lst.append([i, j, j + 100, predicted])
else: 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