def process_class(cls, logs_dir): print("Processing %s" % cls) logs=filter(lambda arg: cls in arg, os.listdir(logs_dir)) for log in logs: signal=Signal.read_3d_csv(os.path.join(logs_dir, log)) chunk=walking_chunk index, dist=use_dtw(signal, chunk) print("Log file %s: DTW dist %f" % (log, dist))
def main(): parser=ArgumentParser() parser.add_argument("--log", "-l", required=True) parser.add_argument("--output", "-o", required=True) args=parser.parse_args() signal=Signal.read_3d_csv(args.log) print("Length: %d" % len(signal.time)) time_diffs=np.diff(signal.time) print("dt: %f..%f" % (min(time_diffs), max(time_diffs))) cutter=LogCutter(signal) working_figure=plt.gcf() display_signal(signal, False) #print(working_figure.number) cutter.connect(working_figure) plt.show() subsignal=Signal.get_subsignal(signal, cutter.left_bound, cutter.right_bound) Signal.save_csv(subsignal, args.output)
#! /usr/bin/python3 from core.signal import Signal import _ucrdtw as dtw import os walking_chunk=Signal.read_3d_csv("parse/chunks/walking_chunk.txt") def process_class(cls, logs_dir): print("Processing %s" % cls) logs=filter(lambda arg: cls in arg, os.listdir(logs_dir)) for log in logs: signal=Signal.read_3d_csv(os.path.join(logs_dir, log)) chunk=walking_chunk index, dist=use_dtw(signal, chunk) print("Log file %s: DTW dist %f" % (log, dist)) def use_dtw(signal, chunk): signal_x=signal.axes_data[0] chunk_x=chunk.axes_data[0] index, dist = dtw.ucrdtw(signal_x, chunk_x, 0.05, False) return index, dist
#! /usr/bin/python3 from argparse import ArgumentParser from core.signal import Signal, display_signal import numpy as np import matplotlib matplotlib.use('TkAgg') import pylab as plt parser=ArgumentParser() parser.add_argument("--log", "-l", required=True) parser.add_argument("--show", "-s", required=False, action="store_true") args=parser.parse_args() signal=Signal.read_3d_csv(args.log) print("Length: %d" % len(signal.time)) time_diffs=np.diff(signal.time) print("dt: %f..%f" % (min(time_diffs), max(time_diffs))) if args.show: display_signal(signal, True)