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

Allow interp factor below 2 for RIFE NCNN VS

Browse Source
This commit is contained in:
n00mkrad
2026-01-29 00:36:12 +01:00
parent 02fced4503
commit a665a4300e
3 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
CodeLegacy/Data/AiInfo.cs
View File

@@ -17,6 +17,7 @@ namespace Flowframes.Data
public string PkgDir { get { return NameInternal.Replace("_", "-").Lower(); } }
public enum InterpFactorSupport { Fixed, AnyPowerOfTwo, AnyInteger, AnyFloat }
public InterpFactorSupport FactorSupport { get; set; } = InterpFactorSupport.Fixed;
public bool AllowLessThan2xFactor { get; set; } = false;
public int[] SupportedFactors { get; set; } = new int[0];
public bool Piped { get; set; } = false;

1
CodeLegacy/Data/Implementations.cs
View File

@@ -42,6 +42,7 @@ namespace Flowframes.Data
NameLong = "Real-Time Intermediate Flow Estimation",
FactorSupport = AiInfo.InterpFactorSupport.AnyFloat,
SupportedFactors = Enumerable.Range(2, 15).ToArray(), // 2 to 16
AllowLessThan2xFactor = true,
Piped = true
};

20
CodeLegacy/Ui/MainUiFunctions.cs
View File

@@ -171,7 +171,7 @@ namespace Flowframes.Ui
if (ai.NameInternal == Implementations.rifeNcnn.NameInternal && !Program.mainForm.GetModel(ai).Dir.Contains("v4"))
{
if (factor != 2)
Logger.Log($"{ai.FriendlyName} models before 4.0 only support 2x interpolation!");
Logger.Log($"RIFE NCNN models before 4.0 only support 2x interpolation!");
return 2;
}
@@ -179,22 +179,30 @@ namespace Flowframes.Ui
if (ai.FactorSupport == AiInfo.InterpFactorSupport.Fixed)
{
int closest = ai.SupportedFactors.Min(i => (Math.Abs(factor.RoundToInt() - i), i)).i;
return (float)closest;
return closest;
}
if(ai.FactorSupport == AiInfo.InterpFactorSupport.AnyPowerOfTwo)
float min = ai.AllowLessThan2xFactor ? 1f : 2f;
if(factor < min)
{
return ToNearestPow2(factor.RoundToInt()).Clamp(2, 128);
Logger.Log($"The selected AI model does not support interpolation factors below {min}x.");
return factor = min;
}
if (ai.FactorSupport == AiInfo.InterpFactorSupport.AnyPowerOfTwo)
{
return ToNearestPow2(factor.RoundToInt()).Clamp(min.RoundToInt(), 128);
}
if(ai.FactorSupport == AiInfo.InterpFactorSupport.AnyInteger)
{
return factor.RoundToInt().Clamp(2, 128);
return factor.RoundToInt().Clamp(min.RoundToInt(), 128);
}
if(ai.FactorSupport == AiInfo.InterpFactorSupport.AnyFloat)
{
return factor.Clamp(2, 128);
return factor.Clamp(min, 128);
}
return factor;