def algodir(basedir):
try:
os.makedirs(rdir(os.path.join(basedir, "algo")))
except OSError:
pass
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):
