2021-08-31 00:09:47 +02:00
|
|
|
|
using System.Collections.Generic;
|
2021-08-23 16:50:18 +02:00
|
|
|
|
|
|
|
|
|
|
namespace Flowframes.Data
|
|
|
|
|
|
{
|
|
|
|
|
|
class Implementations
|
|
|
|
|
|
{
|
2022-05-31 22:17:22 +02:00
|
|
|
|
public static AI rifeCuda = new AI(AI.AiBackend.Pytorch, "RIFE_CUDA", "RIFE",
|
|
|
|
|
|
"CUDA/Pytorch Implementation of RIFE (Nvidia Only!)", "rife-cuda", AI.InterpFactorSupport.AnyInteger, new int[] { 2, 3, 4, 5, 6, 7, 8, 9, 10 });
|
2021-08-31 00:09:47 +02:00
|
|
|
|
|
2022-06-01 16:23:00 +02:00
|
|
|
|
public static AI rifeNcnnVs = new AI(AI.AiBackend.Ncnn, "RIFE_NCNN_VS", "RIFE (NCNN/VS)",
|
|
|
|
|
|
"Vulkan/NCNN/VapourSynth Implementation of RIFE", "rife-ncnn-vs", AI.InterpFactorSupport.AnyFloat, new int[] { 2, 3, 4, 5, 6, 7, 8, 9, 10 }) { Piped = true };
|
|
|
|
|
|
|
2022-05-31 22:17:22 +02:00
|
|
|
|
public static AI rifeNcnn = new AI(AI.AiBackend.Ncnn, "RIFE_NCNN", "RIFE (NCNN)",
|
|
|
|
|
|
"Vulkan/NCNN Implementation of RIFE", "rife-ncnn", AI.InterpFactorSupport.AnyFloat, new int[] { 2, 3, 4, 5, 6, 7, 8, 9, 10 });
|
2021-08-31 00:09:47 +02:00
|
|
|
|
|
2022-05-31 22:17:22 +02:00
|
|
|
|
public static AI flavrCuda = new AI(AI.AiBackend.Pytorch, "FLAVR_CUDA", "FLAVR",
|
|
|
|
|
|
"Experimental Pytorch Implementation of FLAVR (Nvidia Only!)", "flavr-cuda", AI.InterpFactorSupport.Fixed, new int[] { 2, 4, 8 });
|
2021-08-31 00:09:47 +02:00
|
|
|
|
|
2022-05-31 22:17:22 +02:00
|
|
|
|
public static AI dainNcnn = new AI(AI.AiBackend.Ncnn, "DAIN_NCNN", "DAIN (NCNN)",
|
|
|
|
|
|
"Vulkan/NCNN Implementation of DAIN", "dain-ncnn", AI.InterpFactorSupport.AnyFloat, new int[] { 2, 3, 4, 5, 6, 7, 8 });
|
2021-08-23 16:50:18 +02:00
|
|
|
|
|
2022-05-31 22:17:22 +02:00
|
|
|
|
public static AI xvfiCuda = new AI(AI.AiBackend.Pytorch, "XVFI_CUDA", "XVFI",
|
|
|
|
|
|
"CUDA/Pytorch Implementation of XVFI (Nvidia Only!)", "xvfi-cuda", AI.InterpFactorSupport.AnyInteger, new int[] { 2, 3, 4, 5, 6, 7, 8, 9, 10 });
|
|
|
|
|
|
|
2022-06-04 12:43:48 +02:00
|
|
|
|
public static List<AI> networks = new List<AI> { rifeCuda, rifeNcnnVs, rifeNcnn, flavrCuda, dainNcnn, xvfiCuda };
|
2021-08-23 16:50:18 +02:00
|
|
|
|
|
|
|
|
|
|
public static AI GetAi (string aiName)
|
|
|
|
|
|
{
|
|
|
|
|
|
foreach(AI ai in networks)
|
|
|
|
|
|
{
|
2022-05-31 22:17:22 +02:00
|
|
|
|
if (ai.AiName == aiName)
|
2021-08-23 16:50:18 +02:00
|
|
|
|
return ai;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
2022-06-09 00:30:58 +02:00
|
|
|
|
Logger.Log($"AI implementation lookup failed! This should not happen! Please tell the developer! (Implementations.cs)");
|
2021-08-23 16:50:18 +02:00
|
|
|
|
return networks[0];
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|