Mirrors/flowframes
1
0
Fork 0
mirror of https://github.com/n00mkrad/flowframes.git synced 2025-12-23 19:59:31 +01:00
Code Issues Packages Projects Releases Wiki Activity

RIFE-CUDA image piping tests [WIP]

Browse Source
This commit is contained in:
N00MKRAD
2021-02-27 12:46:25 +01:00
parent 8699a5a7fb
commit f30ae16ce7
1 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
Code/OS/AiProcess.cs
View File

@@ -14,6 +14,7 @@ using Flowframes.Data;
using Flowframes.MiscUtils; using Flowframes.MiscUtils;
using Flowframes.Magick; using Flowframes.Magick;
using Flowframes.Media; using Flowframes.Media;
using System.Drawing;
namespace Flowframes namespace Flowframes
{ {
@@ -135,7 +136,7 @@ namespace Flowframes
if (!OSUtils.ShowHiddenCmd()) if (!OSUtils.ShowHiddenCmd())
{ {
rifePy.OutputDataReceived += (sender, outLine) => { LogOutput("[O] " + outLine.Data, "rife-cuda-log"); }; rifePy.OutputDataReceived += (sender, outLine) => { LogOutput(outLine.Data, "rife-cuda-log"); };
rifePy.ErrorDataReceived += (sender, outLine) => { LogOutput("[E] " + outLine.Data, "rife-cuda-log", true); }; rifePy.ErrorDataReceived += (sender, outLine) => { LogOutput("[E] " + outLine.Data, "rife-cuda-log", true); };
} }
rifePy.Start(); rifePy.Start();
@@ -288,6 +289,26 @@ namespace Flowframes
if (string.IsNullOrWhiteSpace(line) || line.Length < 6) if (string.IsNullOrWhiteSpace(line) || line.Length < 6)
return; return;
Stopwatch sw = new Stopwatch();
sw.Restart();
if (line.Contains("iVBOR"))
{
try
{
string[] split = line.Split(':');
//MemoryStream stream = new MemoryStream(Convert.FromBase64String(split[1]));
//Image img = Image.FromStream(stream);
Logger.Log($"Received image {split[0]} in {sw.ElapsedMilliseconds} ms", true);
}
catch (Exception e)
{
Logger.Log($"Failed to decode b64 string - {e}:");
Logger.Log(line);
}
return;
}
Logger.LogToFile(line, false, logFilename); Logger.LogToFile(line, false, logFilename);
if (line.Contains("ff:nocuda-cpu")) if (line.Contains("ff:nocuda-cpu"))
@@ -376,7 +397,7 @@ namespace Flowframes
static string RifeNcnn2Workaround (string modelName, bool reset = false) static string RifeNcnn2Workaround (string modelName, bool reset = false)
{ {
if (modelName != "RIFE20") return modelName; if (!modelName.StartsWith("RIFE2")) return modelName;
string validMdlName = "rife-v2"; string validMdlName = "rife-v2";
string rifeFolderPath = Path.Combine(Paths.GetPkgPath(), Path.GetFileNameWithoutExtension(Packages.rifeNcnn.fileName)); string rifeFolderPath = Path.Combine(Paths.GetPkgPath(), Path.GetFileNameWithoutExtension(Packages.rifeNcnn.fileName));
string modelFolderPath = Path.Combine(rifeFolderPath, modelName); string modelFolderPath = Path.Combine(rifeFolderPath, modelName);