53 lines
1.3 KiB
C#
53 lines
1.3 KiB
C#
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.ComponentModel;
|
|||
|
|
using System.Linq;
|
|||
|
|
using System.Reflection;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Threading.Tasks;
|
|||
|
|
|
|||
|
|
namespace DHSoftware
|
|||
|
|
{
|
|||
|
|
public static class EnumHelper
|
|||
|
|
{
|
|||
|
|
public enum SelectPicType
|
|||
|
|
{
|
|||
|
|
[Description("训练图片")]
|
|||
|
|
train =0,
|
|||
|
|
[Description("测试图片")]
|
|||
|
|
test
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
public enum NetModel
|
|||
|
|
{
|
|||
|
|
[Description("目标检测")]
|
|||
|
|
training = 0,
|
|||
|
|
[Description("语义分割")]
|
|||
|
|
train_seg,
|
|||
|
|
[Description("模型导出")]
|
|||
|
|
emport,
|
|||
|
|
[Description("推理预测")]
|
|||
|
|
infernce
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
public static string GetEnumDescription(this Enum enumObj)
|
|||
|
|
{
|
|||
|
|
Type t = enumObj.GetType();
|
|||
|
|
FieldInfo f = t.GetField(enumObj.ToString());
|
|||
|
|
if (f == null)
|
|||
|
|
{
|
|||
|
|
return enumObj.ToString();
|
|||
|
|
}
|
|||
|
|
DescriptionAttribute attr = f.GetCustomAttribute<DescriptionAttribute>();
|
|||
|
|
if (attr != null)
|
|||
|
|
{
|
|||
|
|
return attr.Description;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
return enumObj.ToString();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|