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); }
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); }