def tf_freeze_graph(sess: Session,
saver: Saver,
model_name: str,
logdir: str,
model_suffix: str = "pb",
conf_suffix: str = "pbtxt",
checkpoint_suffix: str = "ckpt"):
"""
python freeze_graph.py --output_graph =./pbs/frozenGraph.pb --output_node_names=genderOut,ageOut --input_binary=true
"""
input_conf = write_graph(
sess.graph_def, logdir,
extsep.join('_'.join((model_name, "graphDef")), conf_suffix))
input_binary = True
input_graph = write_graph(sess.graph_def,
logdir,
extsep.join('_'.join((model_name, "graphDef")),
model_suffix),
as_text=False)
input_checkpoint = saver.save(
sess, path_join(logdir, extsep.join(model_name, checkpoint_suffix)))
output_node_names = ""
output_graph = path_join(
logdir,
extsep.join('_'.join((model_name, "frozenGraphDef")), model_suffix))
freeze_graph(input_graph, "", input_binary, input_checkpoint,
output_node_names, "save/restore_all", "save/Const:0",
output_graph, True, "")
return output_graph
def plugin_album(self, songs):
songs.sort(lambda a, b: cmp(a('~#track'), b('~#track')) or
cmp(a('~basename'), b('~basename')) or
cmp(a, b))
chooser = filechooser(save=True, title=songs[0]('album'))
resp = chooser.run()
fn = chooser.get_filename()
chooser.destroy()
if resp != Gtk.ResponseType.ACCEPT:
return
base, ext = splitext(fn)
if not ext:
fn = extsep.join([fn, 'tags'])
global lastfolder
lastfolder = dirname(fn)
out = open(fn, 'w')
for song in songs:
print>>out, str(song('~basename'))
keys = song.keys()
keys.sort()
for key in keys:
if key.startswith('~'):
continue
for val in song.list(key):
print>>out, '%s=%s' % (key, val.encode('utf-8'))
print>>out
def plugin_album(self, songs):
songs.sort(lambda a, b: cmp(a('~#track'), b('~#track')) or
cmp(a('~basename'), b('~basename')) or
cmp(a, b))
chooser = filechooser(save=True, title=songs[0]('album'))
resp = chooser.run()
fn = chooser.get_filename()
chooser.destroy()
if resp != Gtk.ResponseType.ACCEPT:
return
base, ext = splitext(fn)
if not ext:
fn = extsep.join([fn, 'tags'])
global lastfolder
lastfolder = dirname(fn)
out = open(fn, 'w')
for song in songs:
print>>out, str(song('~basename'))
keys = song.keys()
keys.sort()
for key in keys:
if key.startswith('~'):
continue
for val in song.list(key):
print>>out, '%s=%s' % (key, val.encode('utf-8'))
print>>out
def get_abs_template_path(template_name, directory, extension):
""" Given a template name, a directory, and an extension,
return the absolute path to the template. """
# Get the relative path
relative_path = join(directory, template_name)
file_with_ext = template_name
if extension:
# If there is a default extension, but no file extension, then add it
file_name, file_ext = splitext(file_with_ext)
if not file_ext:
file_with_ext = extsep.join((file_name, extension.replace(extsep, '')))
# Rebuild the relative path
relative_path = join(directory, file_with_ext)
return abspath(relative_path)
def __init__(self, filename, include_width_in_name=False,
target=None, motif_file='unknown_motif', genome=None,
*args, **kwargs):
fext = splitext(filename)[1].lstrip(extsep)
if fext == 'bed':
self.is_bed = True
self.is_xls = False
elif fext == 'xls':
self.is_bed = False
self.is_xls = True
else: raise InvalidFileException
motif_name = sub('\W', '_', abspath(motif_file))
target = target + sep + motif_name
super(PeaksFilenameParser, self).__init__(filename,
target = target,
*args, **kwargs)
self.fasta_file = None
for file_extension in ['fa', 'fasta', 'FA', 'FASTA']:
fasta_file = join(self.input_dir,
extsep.join([self.protoname, file_extension]))
debug("Trying", fasta_file)
if exists(fasta_file):
self.fasta_file = fasta_file
debug("Using", fasta_file)
break
if self.fasta_file is None:
warning('Could not find the FASTA file for %s',
self.input_file)
if genome is None:
raise Usage("Could not find the FASTA file for ", self.input_file,
" and no genome was specified")
else:
t = try_to_find_genome(genome)
if t is None:
raise Usage("Could not find the FASTA file for ", self.input_file,
" and failed to use %s" % genome)
else:
fasta_file = join(self.input_dir, '%s.fa' % self.protoname)
debug('Creating FASTA file %s for %s using %s',
fasta_file, self.input_file, genome)
input_fhd = open(self.input_file, 'rU')
fasta_fhd = open(fasta_file, 'w')
twobit_reader(t, input_stream=input_fhd,
write=fasta_fhd.write)
fasta_fhd.close()
self.fasta_file = fasta_file
def get_abs_template_path(template_name, directory, extension):
""" Given a template name, a directory, and an extension, return the
absolute path to the template. """
# Get the relative path
relative_path = join(directory, template_name)
file_with_ext = template_name
if extension:
# If there is a default extension, but no file extension, then add it
file_name, file_ext = splitext(file_with_ext)
if not file_ext:
file_with_ext = extsep.join(
(file_name, extension.replace(extsep, '')))
# Rebuild the relative path
relative_path = join(directory, file_with_ext)
return abspath(relative_path)
def main():
ocl_setup()
ovx_setup()
# TODO ogl
ipp_setup()
optimized_setup()
x_train, y_train = get_data()
x_train = x_train.astype(float32)
mlp: ml_ANN_MLP = train_mlp(x_train, list(y_train.tolist()))
time_inference(mlp, x_train)
model_name = "mobilenet_v2_1.0_96"
logdir = path_join(pardir, "data", model_name)
frozen_graph = extsep.join(('_'.join((model_name, "frozen")), "pb"))
net = read_net_from_tf(model_name, logdir, frozen_graph)
sample = x_train[0]
width = uint16(8)
height = uint16(sample.shape[0] / width)
dnn_predict(net, (width, height), sample, sort(unique(y_train)))
def persistence_filename(this_file_name, sub=None):
filename_stem = this_file_name.split(sep)[-1].split(extsep)[0]
qualified_filename_stem = filename_stem if sub is None else '_'.join(
(filename_stem, sub))
return extsep.join((qualified_filename_stem, "h5"))
def convert(filename):
content = pd.read_excel(filename).to_csv()
filename = extsep.join(filename.split(extsep)[:-1]) + extsep + 'csv'
with open(filename, 'w') as f:
f.write(content)
import json
from os import makedirs, environ
from os.path import extsep, isfile, join as path_join
from getpass import getpass
import requests
from appdirs import user_data_dir
from filelock import FileLock
import acumos
from acumos.exc import AcumosError
from acumos.logging import get_logger
from acumos.utils import load_artifact, dump_artifact
_CONFIG_DIR = user_data_dir('acumos')
_CONFIG_PATH = path_join(_CONFIG_DIR, extsep.join(('config', 'json')))
_LOCK_PATH = path_join(_CONFIG_DIR, extsep.join(('config', 'lock')))
_USERNAME_VAR = 'ACUMOS_USERNAME'
_PASSWORD_VAR = 'ACUMOS_PASSWORD'
_TOKEN_VAR = 'ACUMOS_TOKEN'
logger = get_logger(__name__)
gettoken = getpass
def get_jwt(auth_api):
'''Returns the jwt string from config or authentication'''
jwt = environ.get(_TOKEN_VAR)
if not jwt:
config = _configuration()
