Added ECG information (25 mm/sec, 1 mV/10 mm) to plot.
Browse files
app.py
CHANGED
@@ -154,6 +154,7 @@ def predict(numberOfECGs: int = 1,
|
|
154 |
# ====== Create a list of image/label tuples for gradio.Gallery ==========
|
155 |
plotList = []
|
156 |
ecgNumber = 1
|
|
|
157 |
for result in Sessions[request.session_hash].Results:
|
158 |
|
159 |
# ====== Plot ECG =====================================================
|
@@ -166,7 +167,7 @@ def predict(numberOfECGs: int = 1,
|
|
166 |
# ------ ECG-12 -------------------------------------------------------
|
167 |
if ecgType == deepfakeecg.DATA_ECG12:
|
168 |
ecg_plot.plot(result,
|
169 |
-
title = 'ECG-12',
|
170 |
sample_rate = deepfakeecg.ECG_SAMPLING_RATE,
|
171 |
lead_index = [ 'I', 'II', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6', 'III', 'aVR', 'aVL', 'aVF' ],
|
172 |
lead_order = [0, 1, 8, 9, 10, 11, 2, 3, 4, 5, 6, 7],
|
@@ -174,7 +175,7 @@ def predict(numberOfECGs: int = 1,
|
|
174 |
# ------ ECG-8 --------------------------------------------------------
|
175 |
else:
|
176 |
ecg_plot.plot(result,
|
177 |
-
title = 'ECG-8',
|
178 |
sample_rate = deepfakeecg.ECG_SAMPLING_RATE,
|
179 |
lead_index = [ 'I', 'II', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6' ],
|
180 |
lead_order = [0, 1, 2, 3, 4, 5, 6, 7],
|
|
|
154 |
# ====== Create a list of image/label tuples for gradio.Gallery ==========
|
155 |
plotList = []
|
156 |
ecgNumber = 1
|
157 |
+
info = '25 mm/sec, 1 mV/10 mm'
|
158 |
for result in Sessions[request.session_hash].Results:
|
159 |
|
160 |
# ====== Plot ECG =====================================================
|
|
|
167 |
# ------ ECG-12 -------------------------------------------------------
|
168 |
if ecgType == deepfakeecg.DATA_ECG12:
|
169 |
ecg_plot.plot(result,
|
170 |
+
title = 'ECG-12 – ' + info,
|
171 |
sample_rate = deepfakeecg.ECG_SAMPLING_RATE,
|
172 |
lead_index = [ 'I', 'II', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6', 'III', 'aVR', 'aVL', 'aVF' ],
|
173 |
lead_order = [0, 1, 8, 9, 10, 11, 2, 3, 4, 5, 6, 7],
|
|
|
175 |
# ------ ECG-8 --------------------------------------------------------
|
176 |
else:
|
177 |
ecg_plot.plot(result,
|
178 |
+
title = 'ECG-8 – ' + info,
|
179 |
sample_rate = deepfakeecg.ECG_SAMPLING_RATE,
|
180 |
lead_index = [ 'I', 'II', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6' ],
|
181 |
lead_order = [0, 1, 2, 3, 4, 5, 6, 7],
|