348 lines
11 KiB
Go
348 lines
11 KiB
Go
package vqdcms
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"time"
|
|
)
|
|
|
|
type Abnormal struct {
|
|
Value float32 `json:"value"`
|
|
Name string `json:"name"`
|
|
}
|
|
|
|
type abnormal []Abnormal
|
|
|
|
func (r *abnormal) Scan(val interface{}) error {
|
|
b, _ := val.([]byte)
|
|
return json.Unmarshal(b, r)
|
|
}
|
|
|
|
func (r abnormal) Value() (driver.Value, error) {
|
|
return json.Marshal(r)
|
|
}
|
|
|
|
type DefaultValue struct {
|
|
Thr1 float32 `json:"thr1"`
|
|
Name1 string `json:"name1"`
|
|
Thr2 float32 `json:"thr2"`
|
|
Name2 string `json:"name2"`
|
|
Ratio float32 `json:"ratio"` //比例
|
|
}
|
|
|
|
type defaultValue []DefaultValue
|
|
|
|
func (r *defaultValue) Scan(val interface{}) error {
|
|
b, _ := val.([]byte)
|
|
return json.Unmarshal(b, r)
|
|
}
|
|
|
|
func (r defaultValue) Value() (driver.Value, error) {
|
|
return json.Marshal(r)
|
|
}
|
|
|
|
type AbnormalModel struct {
|
|
ID int `gorm:"primary_key" json:"id"`
|
|
ChannelID string `json:"channel_id"`
|
|
ChannelName string `json:"channel_name"`
|
|
AlgoID int `json:"algo_id"`
|
|
AlgoName string `json:"algo_name"`
|
|
TaskName string `json:"task_name"`
|
|
FilePath string `json:"file_path"`
|
|
IsDeep bool `json:"is_deep"`
|
|
Abnormals abnormal `gorm:"type:json" json:"abnormals"`
|
|
DefaultValues defaultValue `gorm:"type:json" json:"default_values"` //设置的默认阈值
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
/* 偏色参数 */
|
|
type VQDColorPara struct {
|
|
ColorThr float32 /* 偏色判断值,*/
|
|
|
|
/* 偏色次数比例,
|
|
默认为0.5, 取值范围: 0~1, 建议范围: 0.1~0.9 */
|
|
ColorAbnNumRatio float32
|
|
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/* 亮暗参数 */
|
|
// OK
|
|
type VQDLgtDarkPara struct {
|
|
/* 超过200的像素比例,过亮阈值,
|
|
默认 0.1, 取值范围: 0~1, 建议范围: 0.1~0.5,
|
|
值设置的越大, 越不灵敏, 就是非常亮才算过亮
|
|
值设置的越小, 越灵敏, 太小的话误报会很多 */
|
|
LightThr float32
|
|
|
|
/* 低于55的像素比例,过暗阈值,
|
|
默认 0.4, 取值范围: 0~1, 建议范围: 0.2~0.6
|
|
值设置的越大, 越不灵敏, 就是非常暗才算过暗
|
|
值设置的越小, 越灵敏, 太小的话误报会很多 */
|
|
DarkThr float32
|
|
|
|
/* 偏暗或者偏亮次数比例, 连续多次检测出现的次数比例
|
|
默认为0.5, 取值范围: 0~1, 建议范围: 0.1~0.9 */
|
|
LgtDarkAbnNumRatio float32
|
|
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/* 清晰度参数 */
|
|
// OK
|
|
type VQDClarityPara struct {
|
|
/* 清晰度判断阈值, 小于该值判定为模糊,
|
|
默认为0.4, 取值范围: 0~1, 建议范围: 0.3~0.99
|
|
值设置的越大, 越不灵敏, 太大的话误报模糊会很多
|
|
值设置的越小, 就是非常模糊才算模糊 */
|
|
ClarityThr float32
|
|
|
|
/* 清晰度异常次数比例,
|
|
默认为0.5, 取值范围: 0~1, 建议范围: 0.1~0.9 */
|
|
ClarityAbnNumRatio float32
|
|
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/* 噪声参数 */
|
|
type VQDNoisePara struct {
|
|
NoiseThr float32 /* 噪声判断阈值,默认为 35 */
|
|
|
|
/* 噪声次数比例,
|
|
默认为0.5, 取值范围: 0~1, 建议范围: 0.1~0.9 */
|
|
NoiseAbnNumRatio float32
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/* 对比度参数 */
|
|
type VQDContrastPara struct {
|
|
CtraLowThr float32 /* 低对比度判断阈值,默认为 30 */
|
|
CtraHighThr float32 /* 高对比度判断阈值,默认为 120 */
|
|
|
|
/* 对比度异常次数比例,
|
|
默认为0.5, 取值范围: 0~1, 建议范围: 0.1~0.9 */
|
|
CtraAbnNumRatio float32
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/* 遮挡参数 */
|
|
type VQDOcclusionPara struct {
|
|
OcclusionThr float32 /* 遮挡判断阈值,默认为 6 */
|
|
|
|
/* 遮挡次数比例
|
|
默认为0.5, 取值范围: 0~1, 建议范围: 0.1~0.9 */
|
|
OcclusionAbnNumRatio float32
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/* 蓝屏参数 */
|
|
// OK
|
|
type VQDBluePara struct {
|
|
/* 蓝屏判断阈值, 蓝色像素占图像面积比例,
|
|
默认为 0.6, 取值范围: 0~1, 建议范围 0.4~0.9 */
|
|
BlueThr float32
|
|
|
|
/* 蓝屏次数比例, 连续多帧中蓝色占比
|
|
默认为0.5, 取值范围: 0~1, 建议范围: 0.1~0.9 */
|
|
BlueAbnNumRatio float32
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/* 抖动参数, 做抖动检测, 尽量隔帧取帧做, */
|
|
// OK
|
|
type VQDSharkPara struct {
|
|
/* 抖动阈值参数
|
|
全局运动均值, 无论是目标移动还是画面抖动, 全局运动均值相对来讲都会比较大,
|
|
默认为 0.2, 取值范围: 0~10, 建议范围: 0.1~0.8 */
|
|
SharkThr float32
|
|
|
|
/* 出现抖动次数的比例,
|
|
默认为0.2, 取值范围: 0~1, 建议范围: 0.1~0.6 */
|
|
SharkAbnNumRatio float32
|
|
|
|
/* 全局运动幅度方差阈值, 理论上来讲, 抖动为全局画面移动, 方向幅度具有一致性, 方差应较小
|
|
默认为0.2, 取值范围: 0~1, 建议范围: 0.05~0.5
|
|
如果内部有很多运动目标, 该值较大, 所以设置较大, 会使得在有运动目标时, 判定为抖动 */
|
|
Resverd1 float32
|
|
|
|
/* 全局运动方向方差阈值, 小于该值时, 判定为全局运动, 可能会抖动
|
|
默认为0.2, 取值范围: 0~1, 建议范围: 0.05~0.5
|
|
如果内部有很多运动目标, 该值较大, 所以设置较大, 会使得在有运动目标时, 判定为抖动 */
|
|
Resverd2 float32
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/* 冻结参数 */
|
|
// OK
|
|
type VQDFreezePara struct {
|
|
/* 冻结阈值参数
|
|
静止的像素比例, 默认为0.999, 取值范围: 0.8~1, 建议范围: 0.99~1
|
|
如果画面静止, 该值比较大, 甚至会接近于1, */
|
|
FreezeThr float32 /* 冻结阈值 */
|
|
|
|
/* 冻结帧数占得比例, 尽量设置的大一些, 避免那些无运动目标的静止监控画面
|
|
默认为0.99, 取值范围: 0.8~1, 建议范围: 0.95~1, */
|
|
FreezeAbnNumRatio float32
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/*
|
|
马赛克检测
|
|
|
|
NEW: 新增深度学习版本 2.0.0
|
|
*/
|
|
type VQDMosaicPara struct {
|
|
/* 马赛克阈值参数
|
|
NEW: 新增深度学习版本 20240717
|
|
默认为 0.1 取值范围: 0~1, 建议范围: 0.1~0.9,
|
|
大于该值判定为偏色, 小于该值判定为正常,
|
|
*/
|
|
MosaicThr float32
|
|
|
|
/* 马赛克次数比例, 连续多帧中马赛克占比
|
|
默认为0.5,取值范围: 0~1, 建议范围: 0.3
|
|
连续多帧中马赛克占比 */
|
|
MosaicAbnNumRatio float32
|
|
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
/*
|
|
花屏检测 NEW NEW NEW NEW NEW
|
|
|
|
NEW: 新增深度学习版本 2.0.0
|
|
*/
|
|
type VQDFlowerPara struct {
|
|
/* 花屏阈值参数
|
|
NEW: 新增深度学习版本 20240717
|
|
默认为 0.1 取值范围: 0~1, 建议范围: 0.1~0.9,
|
|
大于该值判定为偏色, 小于该值判定为正常,
|
|
*/
|
|
FlowerThr float32
|
|
|
|
/* 花屏次数比例, 连续多帧中花屏占比
|
|
默认为0.5, 取值范围: 0~1, 建议范围: 0.3
|
|
连续多帧中花屏占比 */
|
|
FlowerAbnNumRatio float32
|
|
|
|
Resverd1 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 float32 /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 float32 /* 保留,暂时不设置,默认 -999 */
|
|
}
|
|
|
|
type VQDPara struct {
|
|
VecFrmNum int /* 连续分析帧数(2-64), 默认为10, 最大为 64 */
|
|
UseDeepLearning bool /* 是否使用深度学习版本, 默认使用深度学习版本 NEW: 20240717 */
|
|
EnableFunc int /* 功能使能,默认为 NXU_VQD_ENABLE_ALL,多项功能用或 */
|
|
ColorPara VQDColorPara /* 偏色参数 */
|
|
LgtDarkPara VQDLgtDarkPara /* 亮暗参数 */
|
|
ClarityPara VQDClarityPara /* 清晰度参数 */
|
|
NoisePara VQDNoisePara /* 噪声参数 */
|
|
ContrastPara VQDContrastPara /* 对比度参数 */
|
|
OcclusionPara VQDOcclusionPara /* 遮挡参数 */
|
|
BluePara VQDBluePara /* 蓝屏参数 */
|
|
SharkPara VQDSharkPara /* 抖动参数 */
|
|
FreezePara VQDFreezePara /* 冻结参数 */
|
|
|
|
MosaicPara VQDMosaicPara /* 马赛克检测 */
|
|
FlowerPara VQDFlowerPara /* 花屏检测 */
|
|
}
|
|
|
|
// config cvrdo.VQDAlgoConfig
|
|
func NewVQDPara() VQDPara {
|
|
return VQDPara{
|
|
VecFrmNum: 2,
|
|
UseDeepLearning: true,
|
|
EnableFunc: GetAlgoEnable([]string{"vqd_config", "vqd_lgt_dark", "vqd_blue", "vqd_clarity", "vqd_shark", "vqd_freeze",
|
|
"vqd_color", "vqd_occlusion", "vqd_noise", "vqd_contrast", "vqd_mosaic", "vqd_flower"}),
|
|
ColorPara: VQDColorPara{
|
|
ColorThr: 0.3,
|
|
ColorAbnNumRatio: 0.6,
|
|
},
|
|
LgtDarkPara: VQDLgtDarkPara{
|
|
LightThr: 0.3,
|
|
DarkThr: 0.4,
|
|
LgtDarkAbnNumRatio: 0.6,
|
|
},
|
|
ClarityPara: VQDClarityPara{
|
|
ClarityThr: 0.5,
|
|
ClarityAbnNumRatio: 0.6,
|
|
},
|
|
NoisePara: VQDNoisePara{
|
|
NoiseThr: 0.3,
|
|
NoiseAbnNumRatio: 0.6,
|
|
},
|
|
ContrastPara: VQDContrastPara{
|
|
CtraLowThr: 0.3,
|
|
CtraHighThr: 0.8,
|
|
CtraAbnNumRatio: 0.6,
|
|
},
|
|
OcclusionPara: VQDOcclusionPara{
|
|
OcclusionThr: 0.1,
|
|
OcclusionAbnNumRatio: 0.6,
|
|
},
|
|
BluePara: VQDBluePara{
|
|
BlueThr: 0.3,
|
|
BlueAbnNumRatio: 0.6,
|
|
},
|
|
SharkPara: VQDSharkPara{
|
|
SharkThr: 0.2,
|
|
SharkAbnNumRatio: 0.2,
|
|
},
|
|
FreezePara: VQDFreezePara{
|
|
FreezeThr: 0.999,
|
|
FreezeAbnNumRatio: 0.99,
|
|
},
|
|
MosaicPara: VQDMosaicPara{
|
|
MosaicThr: 0.3,
|
|
MosaicAbnNumRatio: 0.6,
|
|
},
|
|
FlowerPara: VQDFlowerPara{
|
|
FlowerThr: 0.3,
|
|
FlowerAbnNumRatio: 0.6,
|
|
},
|
|
}
|
|
}
|
|
|
|
/*
|
|
异常结果
|
|
|
|
当前各异常返回出来的值,作为参考,外面也可以存到列表中进行应用
|
|
*/
|
|
type VQDResult struct {
|
|
AbnormalType int /* 异常类型,多种类型的,利用与或来确定异常类型 */
|
|
ColorDev float32 /* 偏色检测 */
|
|
LgtDark float32 /* 亮暗检测 */
|
|
Clarity float32 /* 清晰度检测 */
|
|
Noise float32 /* 噪声检测 */
|
|
Contrast float32 /* 对比度检测 */
|
|
Occlusion float32 /* 遮挡检测 */
|
|
Blue float32 /* 蓝屏检测 */
|
|
Shark float32 /* 抖动检测 */
|
|
Freeze float32 /* 冻结检测 */
|
|
Mosaic float32 /* 马赛克检测 NEW: 新增深度学习版本 2.0.0*/
|
|
Flower float32 /* 花屏检测 NEW: 新增深度学习版本 2.0.0*/
|
|
Resverd1 int /* 保留,暂时不设置,默认 -999 */
|
|
Resverd2 int /* 保留,暂时不设置,默认 -999 */
|
|
Resverd3 int /* 保留,暂时不设置,默认 -999 */
|
|
}
|