def algodir(basedir): try: os.makedirs(rdir(os.path.join(basedir, "algo"))) except OSError: pass
d[frame_no]['y'] = real_y d[frame_no]['confidence'] = conf return d @transform(ddir("*/positions.npy"), regex(r".+/(.+)/positions.npy$"), [os.path.join(REPORT_DIR, r"\1", "truth.npy")], r"\1" ) def get_truth(positions_file, (output_file, ), basedir): truth_data = truth(ddir(basedir)) try: os.makedirs(rdir(basedir)) except OSError: pass np.save(output_file, truth_data) def algodir(basedir): try: os.makedirs(rdir(os.path.join(basedir, "algo"))) except OSError: pass @transform(ddir("*/positions.npy"), regex(r".+/(.+)/positions.npy$"), [os.path.join(REPORT_DIR, r"\1", "algo", "current.npy")], r"\1"
fs = organizedata.get_frames(basedir, frame_subset) for fi, frame_no in enumerate(frame_subset): real_x, real_y, conf = func(fs[fi], env, **config) d[frame_no]['x'] = real_x d[frame_no]['y'] = real_y d[frame_no]['confidence'] = conf return d @transform(ddir("*/positions.npy"), regex(r".+/(.+)/positions.npy$"), [os.path.join(REPORT_DIR, r"\1", "truth.npy")], r"\1") def get_truth(positions_file, (output_file, ), basedir): truth_data = truth(ddir(basedir)) try: os.makedirs(rdir(basedir)) except OSError: pass np.save(output_file, truth_data) def algodir(basedir): try: os.makedirs(rdir(os.path.join(basedir, "algo"))) except OSError: pass @transform(ddir("*/positions.npy"), regex(r".+/(.+)/positions.npy$"), [os.path.join(REPORT_DIR, r"\1", "algo", "current.npy")], r"\1") def get_algo_current(positions_file, (output_file, ), basedir):