def agum_feats(in_path, n_epochs=1000):
paths = files.prepare_dirs(in_path, "agum",
["spline", "agum_seqs", "models", "feats"])
spline.ens_upsample(in_path, paths["spline"])
files.make_dir(paths["agum_seqs"])
agum.ens_agum(paths["spline"], paths["agum_seqs"])
basic.ts.ens_train(paths["agum_seqs"], paths["models"], n_epochs)
basic.ts.ens_extract(paths["spline"], paths["models"], paths["feats"])
def single_exp_template(in_path,
out_path,
train,
extract,
size=64,
n_epochs=1000):
paths = files.prepare_dirs(out_path, ["spline", "nn", "feats"])
paths['seqs'] = in_path
spline.upsample(paths['seqs'], paths['spline'], size)
train(paths["spline"], paths["nn"], n_epochs=n_epochs)
extract(paths["spline"], paths["nn"], paths["feats"])
def simple_exp(in_path, out_path, n_epochs=1000):
paths = files.prepare_dirs(out_path, ["nn", "feats"])
paths['frames'] = in_path
train, extract = get_train()
train(paths["frames"], paths["nn"], n_epochs=n_epochs)
extract(paths["frames"], paths["nn"], paths["feats"])
def stats_feats(in_path):
paths = files.prepare_dirs(in_path, "stats", ["feats"])
stats.ens_stats(in_path, paths["feats"])
def full_seqs(in_path, n_epochs=1000):
paths = files.prepare_dirs(in_path, None, ["frame_model", "seqs"])
basic.full_train(in_path, paths["frame_model"], n_epochs)
files.make_dir(path["seqs"])
basic.extract(model_path, seq_path, in_path)
def basic_seqs(in_path, n_epochs=1000):
paths = files.prepare_dirs(in_path, None, ["frame_models", "seqs"])
basic.ens_train(in_path, paths["frame_models"], n_epochs)
basic.ens_extract(in_path, paths["frame_models"], paths["seqs"])
def sim_seqs(in_path, n_epochs=350):
paths = files.prepare_dirs(in_path, None, ["frame_models", "seqs"])
# frames.ens_train(in_path,paths["frame_models"],n_epochs)
frames.ens_extract(in_path, paths["frame_models"], paths["seqs"])
def ae_seqs(in_path, n_epochs=1000):
paths = files.prepare_dirs(in_path, None, ["ae", "seqs"])
frames.ae.make_model(in_path, paths["ae"], n_epochs)
frames.ae.extract(in_path, paths["ae"], paths["seqs"])
def sim_feats(in_path, n_epochs=350):
paths = files.prepare_dirs(in_path, "sim", ["spline", "models", "feats"])
spline.ens_upsample(in_path, paths["spline"])
ts.ens_train(paths["spline"], paths["models"], n_epochs)
ts.ens_extract(paths["spline"], paths["models"], paths["feats"])
def basic_feats(in_path, n_epochs=1000):
paths = files.prepare_dirs(in_path, "basic", ["spline", "models", "feats"])
spline.ens_upsample(in_path, paths["spline"], size=512)
basic.ts.ens_train(paths["spline"], paths["models"], n_epochs)
basic.ts.ens_extract(paths["spline"], paths["models"], paths["feats"])