package sdk
|
|
import (
|
"analysis/goconv"
|
"analysis/logo"
|
"analysis/work"
|
"context"
|
"time"
|
|
"basic.com/valib/gosdk.git"
|
)
|
|
// PlateIDDetector plate id
|
type PlateIDDetector struct {
|
config *gosdk.CPlateIDCfg
|
cfgCloud *gosdk.CPlateIDCloudCfg
|
licSrvPath string
|
modelPath string
|
fn func(*gosdk.SDKImage, *gosdk.CRECT, int, int, *work.MsgRS, chan<- work.MsgRS)
|
}
|
|
// NewPlateIDDetector plate
|
func NewPlateIDDetector(licSrv, model string, w, h int) *PlateIDDetector {
|
|
vehicleR := true
|
eparkingR := false
|
cloudR := false
|
// vehicle
|
if vehicleR {
|
// ScenceMode int // 0-顶装,侧装,顶装路径,停车场,公安
|
// ModelMode int //0-不修改,1修改模型
|
return &PlateIDDetector{
|
licSrvPath: licSrv,
|
modelPath: model,
|
fn: vehicle,
|
}
|
}
|
// eparking no model
|
if eparkingR {
|
cfg := gosdk.DefaultPlateIDSDKConfig()
|
cfg.MaxImageWidth, cfg.MaxImageHeight = int32(w), int32(h)
|
cfg.MovingImage = 0
|
cfg.MinHang, cfg.ChangNei = 0, 0
|
|
return &PlateIDDetector{
|
config: cfg,
|
cfgCloud: nil,
|
licSrvPath: licSrv,
|
modelPath: "",
|
fn: eparking,
|
}
|
|
}
|
|
// cloud
|
if cloudR {
|
cfg := gosdk.DefaultPlateIDCloudSDKConfig()
|
|
return &PlateIDDetector{
|
config: nil,
|
cfgCloud: cfg,
|
licSrvPath: licSrv,
|
modelPath: model,
|
fn: cloud,
|
}
|
}
|
return nil
|
}
|
|
// Init impl
|
func (d *PlateIDDetector) Init() bool {
|
if d.config != nil {
|
gosdk.InitPlateIDDetector(d.config, []byte(d.licSrvPath))
|
logo.Infoln("RUN PLATE ID")
|
} else if d.cfgCloud != nil {
|
ret := gosdk.InitPlateIDCloudSDKDetector(d.cfgCloud, []byte(d.licSrvPath), []byte(d.modelPath))
|
if ret != 0 {
|
logo.Errorf("INIT PLATE ID CLOUD SDK ERROR: %d\n", ret)
|
return false
|
}
|
logo.Infoln("RUN CLOUD PLATE ID")
|
} else {
|
ret := gosdk.InitVehicleITSDetector(2, 0, []byte(d.licSrvPath), []byte(d.modelPath))
|
if ret != 0 {
|
logo.Errorf("INIT PLATE ID VEHICLE SDK ERROR: %d\n", ret)
|
return false
|
}
|
logo.Infoln("RUN VEHICLE PLATE ID")
|
}
|
return true
|
}
|
|
// Run impl
|
func (d *PlateIDDetector) Run(ctx context.Context, in <-chan work.MsgRS, out chan<- work.MsgRS, typ string) {
|
tm := time.Now()
|
sc := 0
|
|
for {
|
select {
|
case <-ctx.Done():
|
goconv.Free()
|
return
|
default:
|
rMsg := <-in
|
if !validRemoteMessage(rMsg, typ) {
|
ejectResult(nil, rMsg, out)
|
continue
|
}
|
|
i := unpackImage(rMsg, typ)
|
if i == nil || i.Data == nil || i.Width <= 0 || i.Height <= 0 {
|
ejectResult(nil, rMsg, out)
|
continue
|
}
|
|
imgW, imgH := int(i.Width), int(i.Height)
|
bgrData := goconv.YUV2BGR(i.Data, imgW, imgH)
|
if bgrData == nil {
|
ejectResult(nil, rMsg, out)
|
continue
|
}
|
img := gosdk.SDKImage{Data: bgrData, Width: imgW, Height: imgH}
|
|
rcDetect := gosdk.CRECT{
|
Left: 0,
|
Top: 0,
|
Right: (int32)(imgW),
|
Bottom: (int32)(imgH),
|
}
|
|
mW, mH := 4096, 2160
|
if d.config != nil {
|
mW, mH = int(d.config.MaxImageWidth), int(d.config.MaxImageHeight)
|
}
|
d.fn(&img, &rcDetect, mW, mH, &rMsg, out)
|
|
sc++
|
if sc == 25 {
|
logo.Infoln("PLATE ID DETECTOR RUN 25 FRAME USE TIME: ", time.Since(tm))
|
sc = 0
|
tm = time.Now()
|
}
|
|
if time.Since(tm) > time.Second {
|
logo.Infof("PLATE ID DETECTOR RUN %d FRAME USE TIME: %v", sc, time.Since(tm))
|
sc = 0
|
tm = time.Now()
|
}
|
|
}
|
}
|
}
|
|
func eparking(img *gosdk.SDKImage, rc *gosdk.CRECT, mW, mH int, rMsg *work.MsgRS, out chan<- work.MsgRS) {
|
|
// imgW, imgH := int(img.Width), int(img.Height)
|
|
// fx, fy := 1.0, 1.0
|
|
// if imgW > mW || imgH > mH {
|
// fx, fy = (float64)(mW)/(float64)(imgW), (float64)(mH)/(float64)(imgH)
|
// img.Data = goconv.ResizeBGR(img.Data, imgW, imgH, mW, mH)
|
// if img.Data == nil {
|
// ejectResult(nil, *rMsg, out)
|
// return
|
// }
|
// imgW, imgH = mW, mH
|
// }
|
|
// rImg := gosdk.SDKImage{Data: img.Data, Width: imgW, Height: imgH}
|
// rRc := &gosdk.CRECT{
|
// Left: 0,
|
// Top: 0,
|
// Right: int32(imgW),
|
// Bottom: int32(imgH),
|
// }
|
// result := gosdk.PlateIDDetect(rImg, rRc)
|
// count := len(result)
|
// if count <= 0 {
|
// ejectResult(nil, *rMsg, out)
|
// return
|
// }
|
|
// plateids := convert2ProtoPlateIDResult(result, fx, fy)
|
|
// plateresult := protomsg.PlateIDResult{Result: plateids}
|
// data, err := proto.Marshal(&plateresult)
|
// if err != nil {
|
// logo.Errorln("EPARKING PLATE ID DETECTOR MARSHAL PROTO PLATE IDS ERROR", err)
|
// data = nil
|
// }
|
// ejectResult(data, *rMsg, out)
|
}
|
|
// func convert2ProtoPlateIDResult(obj []gosdk.CPlateIDResult, fx, fy float64) []*protomsg.PlateID {
|
// ret := []*protomsg.PlateID{}
|
|
// for _, v := range obj {
|
|
// rcPlateID := &protomsg.Rect{
|
// Left: (int32)((float64)(v.RcLocation.Left) / fx),
|
// Right: (int32)((float64)(v.RcLocation.Right) / fx),
|
// Top: (int32)((float64)(v.RcLocation.Top) / fy),
|
// Bottom: (int32)((float64)(v.RcLocation.Bottom) / fy),
|
// }
|
// rcLogo := &protomsg.Rect{
|
// Left: (int32)((float64)(v.RcLogoLocation.Left) / fx),
|
// Right: (int32)((float64)(v.RcLogoLocation.Right) / fx),
|
// Top: (int32)((float64)(v.RcLogoLocation.Top) / fy),
|
// Bottom: (int32)((float64)(v.RcLogoLocation.Bottom) / fy),
|
// }
|
// lic := string(v.License[:])
|
// end := len(lic)
|
// for i := len(lic) - 1; i >= 0; i-- {
|
// if lic[i] != '\000' {
|
// end = i + 1
|
// break
|
// }
|
// }
|
// if end > 0 {
|
// lic = lic[:end]
|
// }
|
|
// obj := &protomsg.PlateID{
|
// License: lic,
|
// Color: string(v.Color[:]),
|
// NColor: v.NColor,
|
// NType: v.NType,
|
// NConfidence: v.NConfidence,
|
// NBright: v.NBright,
|
// NDirection: v.NDirection,
|
// RcLocation: rcPlateID,
|
// NTime: v.NTime,
|
// NCarBright: int32(v.NCarBright),
|
// NCarColor: int32(v.NCarColor),
|
// NCarLogo: int32(v.NCarLogo),
|
// NCarType: int32(v.NCarType),
|
// PlateBin: v.PbyPlateBin[:],
|
// NBinPlateWidth: v.NBinPlateWidth[:],
|
// NBinPlateHeight: v.NBinPlateHeight[:],
|
// RcLogoLocation: rcLogo,
|
// NCarModel: v.NCarModel[:],
|
// NCarModelConfidence: v.NCarModelConfidence[:],
|
// }
|
|
// ret = append(ret, obj)
|
// }
|
// return ret
|
// }
|
