2021-08-23 16:50:18 +02:00
|
|
|
|
using Flowframes.IO;
|
|
|
|
|
|
using System;
|
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
|
using System.Linq;
|
|
|
|
|
|
using System.Text;
|
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
|
|
|
|
|
|
|
namespace Flowframes.Data
|
|
|
|
|
|
{
|
|
|
|
|
|
public struct AI
|
|
|
|
|
|
{
|
2021-08-31 00:09:47 +02:00
|
|
|
|
public enum Backend { Pytorch, Ncnn, Tensorflow, Other }
|
|
|
|
|
|
public Backend backend;
|
2021-08-23 16:50:18 +02:00
|
|
|
|
public string aiName;
|
|
|
|
|
|
public string aiNameShort;
|
|
|
|
|
|
public string friendlyName;
|
|
|
|
|
|
public string description;
|
|
|
|
|
|
public string pkgDir;
|
2022-04-06 20:48:47 +02:00
|
|
|
|
public enum FactorSupport { Fixed, AnyPowerOfTwo, AnyInteger, AnyFloat }
|
|
|
|
|
|
public FactorSupport factorSupport;
|
2021-08-23 16:50:18 +02:00
|
|
|
|
public int[] supportedFactors;
|
|
|
|
|
|
|
2022-04-06 20:48:47 +02:00
|
|
|
|
public AI(Backend backend, string aiName, string friendlyName, string desc, string pkgDir, FactorSupport factorSupport = FactorSupport.Fixed, int[] factors = null)
|
2021-08-23 16:50:18 +02:00
|
|
|
|
{
|
2021-08-31 00:09:47 +02:00
|
|
|
|
this.backend = backend;
|
|
|
|
|
|
this.aiName = aiName;
|
|
|
|
|
|
this.aiNameShort = aiName.Split(' ')[0].Split('_')[0];
|
|
|
|
|
|
this.friendlyName = friendlyName;
|
|
|
|
|
|
this.description = desc;
|
|
|
|
|
|
this.pkgDir = pkgDir;
|
|
|
|
|
|
this.supportedFactors = factors;
|
2022-04-06 20:48:47 +02:00
|
|
|
|
this.factorSupport = factorSupport;
|
2021-08-23 16:50:18 +02:00
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|