def cnn_eval_run(*args):
from cnn.eval import run
set_logger_file(
str(
Path(args[1]) /
("eval-" + time.strftime("%Y%m%d%H%M%S") + ".log")))
run(*args)
def cnn_train(args):
from cnn.train import run
run(
args.device,
args.model,
args.topology,
args.version,
args.evalsource,
args.trainsource,
args.threads,
args.prefetch,
args.batch,
args.steps,
args.epochs,
args.ram,
args.initializer,
args.regularizer,
args.constraint,
args.activation,
args.bn,
args.lrn,
args.dropoutrate,
args.labelweight,
args.loss,
args.optimizer,
args.learningrate,
args.learningdecay,
args.verbose,
args.seed
)
def cnn_train_run(*args):
from cnn.train import run
set_logger_file(
str(
Path(args[1]) /
("train-" + time.strftime("%Y%m%d%H%M%S") + ".log")))
run(*args)
def ds_tfrdist(args):
from ds.tfrdist import run
run(
args.source,
args.threads,
args.verbose,
args.seed
)
def ds_genweight(args):
from ds.genweight import run
run(
args.source,
args.expand,
args.ratio,
args.border,
args.preview
)
def ds_cleanmask(args):
from ds.cleanmask import run
run(
args.source,
args.threshold,
args.open,
args.close,
args.minarea,
args.expand,
args.preview
)
def ds_tfrorder(args):
from ds.tfrorder import run
run(
args.source,
args.threads,
args.prefetch,
args.batch,
args.steps,
args.epochs,
args.verbose,
args.seed
)
def cnn_predict(args):
from cnn.predict import run
run(
args.device,
args.model,
args.source,
args.threads,
args.prefetch,
args.batch,
args.epochs,
args.verbose,
args.seed
)
def cnn_eval(args):
from cnn.eval import run
run(
args.device,
args.model,
args.source,
args.threads,
args.prefetch,
args.batch,
args.steps,
args.verbose,
args.seed
)
def cnn_trainmulti(args):
from cnn.trainmulti import run
run(
args.config,
args.model,
args.tensorboard,
args.always,
args.fresh,
args.device,
args.threads,
args.prefetch,
args.ram,
args.verbose
)
def ds_bake(args):
from ds.bake import run
run(
args.device,
args.name,
args.source,
args.compression,
args.batch,
args.testratio,
args.evalratio,
args.size,
args.blur,
args.blurscale,
args.centercrop,
args.scales,
args.flips,
args.rotations,
args.crops,
args.brightness,
args.contrast,
args.gnoise,
args.unoise,
args.seed
)
def ds_preview(args):
from ds.preview import run
run(
args.source
)
def ds_merge2d(args):
from ds.merge2d import run
run(
args.source,
args.destination
)
def ds_jpegorient(args):
from ds.jpegorient import run
run(
args.source
)
def ds_importmve(args):
from ds.importmve import run
run(
args.source,
args.destination
)
def ds_importflower(args):
from ds.importflower import run
run(
args.source,
args.destination
)
