Ejemplo n.º 1
0
        public static string SaveGraphToZip(Session session, ANNModel model, string zippath = null)
        {
            var tempdir = CreateUniqueTempDirectory();

            session.graph.as_default();

            tf.train.Saver().save(session, Path.Combine(tempdir, model.ModelName));

            if (zippath == null)
            {
                zippath = Path.GetTempFileName();
            }

            ZipFile.CreateFromDirectory(tempdir, zippath);

            Directory.Delete(tempdir, true);

            return(zippath);
        }
Ejemplo n.º 2
0
        public static bool SaveGraphToZipStream(Session session, ANNModel model, MemoryStream stream)
        {
            var tempdir = CreateUniqueTempDirectory();

            session.graph.as_default();

            tf.train.Saver().save(session, Path.Combine(tempdir, model.ModelName));

            var zippath = Path.GetTempFileName();

            File.Delete(zippath);

            ZipFile.CreateFromDirectory(tempdir, zippath);

            stream.Position = 0;
            using (FileStream file = new FileStream(zippath, FileMode.Open, FileAccess.Read))
                file.CopyTo(stream);

            Directory.Delete(tempdir, true);

            return(true);
        }