Mirrors/flowframes
1
0
Fork 0
mirror of https://github.com/n00mkrad/flowframes.git synced 2026-09-01 19:51:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

RIFE-CUDA fp16 option and warplayer.py optimization

Browse Source
This commit is contained in:
N00MKRAD
2021-02-21 11:26:18 +01:00
parent 4a673744a5
commit 1d04d2d3ee
7 changed files with 79 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
Code/Forms/SettingsForm.Designer.cs generated
View File

@@ -164,6 +164,8 @@
this.cmdDebugMode = new System.Windows.Forms.ComboBox();
this.titleLabel = new System.Windows.Forms.Label();
this.toolTip1 = new System.Windows.Forms.ToolTip(this.components);
this.label65 = new System.Windows.Forms.Label();
this.rifeCudaFp16 = new System.Windows.Forms.CheckBox();
this.settingsTabList.SuspendLayout();
this.generalTab.SuspendLayout();
this.tabListPage2.SuspendLayout();
@@ -821,6 +823,8 @@
// aiOptsPage
//
this.aiOptsPage.BackColor = System.Drawing.Color.FromArgb(((int)(((byte)(48)))), ((int)(((byte)(48)))), ((int)(((byte)(48)))));
this.aiOptsPage.Controls.Add(this.rifeCudaFp16);
this.aiOptsPage.Controls.Add(this.label65);
this.aiOptsPage.Controls.Add(this.label35);
this.aiOptsPage.Controls.Add(this.panel12);
this.aiOptsPage.Controls.Add(this.dainNcnnTilesize);
@@ -849,7 +853,7 @@
//
this.label35.AutoSize = true;
this.label35.ForeColor = System.Drawing.Color.Silver;
this.label35.Location = new System.Drawing.Point(412, 261);
this.label35.Location = new System.Drawing.Point(412, 291);
this.label35.Margin = new System.Windows.Forms.Padding(10, 10, 10, 7);
this.label35.Name = "label35";
this.label35.Size = new System.Drawing.Size(258, 13);
@@ -860,7 +864,7 @@
//
this.panel12.BackgroundImage = global::Flowframes.Properties.Resources.baseline_create_white_18dp_semiTransparent;
this.panel12.BackgroundImageLayout = System.Windows.Forms.ImageLayout.Zoom;
this.panel12.Location = new System.Drawing.Point(378, 257);
this.panel12.Location = new System.Drawing.Point(378, 287);
this.panel12.Name = "panel12";
this.panel12.Size = new System.Drawing.Size(21, 21);
this.panel12.TabIndex = 78;
@@ -880,7 +884,7 @@
"1024",
"1536",
"2048"});
this.dainNcnnTilesize.Location = new System.Drawing.Point(280, 257);
this.dainNcnnTilesize.Location = new System.Drawing.Point(280, 287);
this.dainNcnnTilesize.Margin = new System.Windows.Forms.Padding(3, 3, 8, 3);
this.dainNcnnTilesize.Name = "dainNcnnTilesize";
this.dainNcnnTilesize.Size = new System.Drawing.Size(87, 21);
@@ -889,7 +893,7 @@
// label27
//
this.label27.AutoSize = true;
this.label27.Location = new System.Drawing.Point(10, 260);
this.label27.Location = new System.Drawing.Point(10, 290);
this.label27.Margin = new System.Windows.Forms.Padding(10, 10, 10, 7);
this.label27.Name = "label27";
this.label27.Size = new System.Drawing.Size(93, 13);
@@ -900,7 +904,7 @@
//
this.label26.AutoSize = true;
this.label26.Font = new System.Drawing.Font("Microsoft Sans Serif", 9.75F, System.Drawing.FontStyle.Bold);
this.label26.Location = new System.Drawing.Point(10, 230);
this.label26.Location = new System.Drawing.Point(10, 260);
this.label26.Margin = new System.Windows.Forms.Padding(10, 10, 10, 7);
this.label26.Name = "label26";
this.label26.Size = new System.Drawing.Size(152, 16);
@@ -1884,6 +1888,25 @@
this.titleLabel.TabIndex = 1;
this.titleLabel.Text = "Settings";
//
// label65
//
this.label65.AutoSize = true;
this.label65.Location = new System.Drawing.Point(11, 210);
this.label65.Margin = new System.Windows.Forms.Padding(10, 10, 10, 7);
this.label65.Name = "label65";
this.label65.Size = new System.Drawing.Size(120, 13);
this.label65.TabIndex = 81;
this.label65.Text = "Fast Mode (CUDA Only)";
//
// rifeCudaFp16
//
this.rifeCudaFp16.AutoSize = true;
this.rifeCudaFp16.Location = new System.Drawing.Point(280, 210);
this.rifeCudaFp16.Name = "rifeCudaFp16";
this.rifeCudaFp16.Size = new System.Drawing.Size(15, 14);
this.rifeCudaFp16.TabIndex = 82;
this.rifeCudaFp16.UseVisualStyleBackColor = true;
//
// SettingsForm
//
this.AutoScaleDimensions = new System.Drawing.SizeF(6F, 13F);
@@ -2058,5 +2081,7 @@
private System.Windows.Forms.ComboBox modelsBaseUrl;
private System.Windows.Forms.Label label42;
private System.Windows.Forms.Panel panel13;
private System.Windows.Forms.CheckBox rifeCudaFp16;
private System.Windows.Forms.Label label65;
}
}

2
Code/Forms/SettingsForm.cs
View File

@@ -84,6 +84,7 @@ namespace Flowframes.Forms
ConfigParser.SaveGuiElement(ncnnGpus);
ConfigParser.SaveGuiElement(ncnnThreads);
ConfigParser.SaveGuiElement(uhdThresh);
ConfigParser.SaveGuiElement(rifeCudaFp16);
ConfigParser.SaveGuiElement(dainNcnnTilesize, ConfigParser.StringMode.Int);
// Video Export
ConfigParser.SaveGuiElement(minOutVidLength, ConfigParser.StringMode.Int);
@@ -135,6 +136,7 @@ namespace Flowframes.Forms
ConfigParser.LoadGuiElement(ncnnGpus);
ConfigParser.LoadGuiElement(ncnnThreads);
ConfigParser.LoadGuiElement(uhdThresh);
ConfigParser.LoadGuiElement(rifeCudaFp16);
ConfigParser.LoadGuiElement(dainNcnnTilesize);
// Video Export
ConfigParser.LoadGuiElement(minOutVidLength);

1
Code/IO/Config.cs
View File

@@ -152,6 +152,7 @@ namespace Flowframes.IO
if (key == "minVidLength") return WriteDefault(key, "2");
// AI
if (key == "uhdThresh") return WriteDefault(key, "1440");
if (key == "rifeCudaFp16") return WriteDefault(key, NvApi.HasTensorCores().ToString());
if (key == "ncnnThreads") return WriteDefault(key, "1");
if (key == "dainNcnnTilesize") return WriteDefault(key, "768");
// Debug / Other / Experimental

3
Code/OS/AiProcess.cs
View File

@@ -120,7 +120,8 @@ namespace Flowframes
string outPath = Path.Combine(inPath.GetParentDir(), outDir);
string wthreads = $"--wthreads {2 * (int)interpFactor}";
string rbuffer = $"--rbuffer {Config.GetInt("rifeCudaBufferSize", 200)}";
string args = $" --input {inPath.Wrap()} --output {outDir} --model {mdl} --exp {(int)Math.Log(interpFactor, 2)} {wthreads} {rbuffer}";
string prec = Config.GetBool("rifeCudaFp16") ? "--fp16" : "";
string args = $" --input {inPath.Wrap()} --output {outDir} --model {mdl} --exp {(int)Math.Log(interpFactor, 2)} {wthreads} {rbuffer} {prec}";
// if (parallel) args = $" --input {inPath.Wrap()} --output {outPath} --model {mdl} --factor {interpFactor}";
// if (parallel) script = "rife-parallel.py";

17
Code/OS/NvApi.cs
View File

@@ -24,7 +24,7 @@ namespace Flowframes.OS
return;
gpu = gpus[0];
Logger.Log($"Initialized NvApi. GPU: {gpu.FullName}");
Logger.Log($"Initialized NvApi. GPU: {gpu.FullName} - Tensor Cores: {HasTensorCores()}");
}
catch (Exception e)
{
@@ -50,7 +50,7 @@ namespace Flowframes.OS
{
return (gpu.MemoryInformation.CurrentAvailableDedicatedVideoMemoryInkB / 1000f / 1024f);
}
catch (Exception e)
catch
{
return 0f;
}
@@ -72,5 +72,18 @@ namespace Flowframes.OS
return "";
}
}
public static bool HasTensorCores ()
{
if (gpu == null)
Init();
if (gpu == null)
return false;
string gpuName = gpu.FullName;
return (gpuName.Contains("RTX 20") || gpuName.Contains("RTX 30") || gpuName.Contains("Tesla V") || gpuName.Contains("Tesla T"));
}
}
}