def process_ground_truth(file_name):
ground_truth = GroundTruth(None, file_name)
ground_truth.process_ground_truth(10)
fig, ax = plt.subplots(2, 1)
fig.suptitle("Time series", fontsize=14)
ax[0].plot(ground_truth.time_base,
ground_truth.ecg_wave,
label='ECG Waveform',
color=(0, 0, 1))
ax[1].plot(ground_truth.time_base,
ground_truth.ecg_heart_rate,
label='ECG Heart rate',
color=(1, 0, 0))
ax[1].plot(ground_truth.time_base,
ground_truth.ecg_average,
label='ECG Heart rate (average)',
color=(0, 1, 0))
ax[0].legend(loc='best')
ax[1].legend(loc='best')
plt.ion()
plt.pause(0.00001)
plt.show()
input("press enter")
def __process_ground_truth(self, video_file_or_camera):
if video_file_or_camera != 0:
# processing a video file, check to see if there is an existing ground_truth file
# noinspection PyPep8
try:
folder = os.path.dirname(video_file_or_camera)
ground_truth_file = "{}/ground_truth.txt".format(folder)
ground_truth = GroundTruth(self.logger, ground_truth_file)
ground_truth.process_ground_truth(
int(self.config["pulse_sample_frames"] /
self.config["video_fps"]))
self.ground_truth = ground_truth
except:
e = sys.exc_info()[0]
self.logger.error("Exception: {} ".format(e))