| | |
| | | import ( |
| | | "basic.com/pubsub/protomsg.git" |
| | | "basic.com/valib/deliver.git" |
| | | "encoding/base64" |
| | | "fmt" |
| | | "github.com/gogo/protobuf/proto" |
| | | "github.com/pierrec/lz4" |
| | | "github.com/pkg/errors" |
| | | "github.com/satori/go.uuid" |
| | | "gocv.io/x/gocv" |
| | | "io/ioutil" |
| | | "mime/multipart" |
| | | "sync" |
| | | "time" |
| | | ) |
| | | |
| | | type FaceSdkService struct { |
| | | File *multipart.File |
| | | File []byte |
| | | Id string |
| | | Result []*protomsg.ResultFaceExtCom |
| | | } |
| | |
| | | Url_Service_PULL = Ipc_Url_Pre + Virtual_FaceSdkId + Ipc_Pull_Ext |
| | | ) |
| | | |
| | | func NewFaceSdkService(file multipart.File) FaceSdkService{ |
| | | func NewFaceSdkService(fileBytes []byte) FaceSdkService{ |
| | | return FaceSdkService{ |
| | | File:&file, |
| | | File:fileBytes, |
| | | Id:uuid.NewV4().String(), |
| | | } |
| | | } |
| | |
| | | } |
| | | |
| | | func (s *FaceSdkService) ReadFromUploadImg() (*protomsg.Image,error){ |
| | | defer (*(s.File)).Close() |
| | | imgB, err := ioutil.ReadAll(*(s.File)) |
| | | if err !=nil{ |
| | | fmt.Println("File.Read err:",err) |
| | | return nil,err |
| | | } |
| | | //defer (*(s.File)).Close() |
| | | //imgB, err := ioutil.ReadAll(*(s.File)) |
| | | //if err !=nil{ |
| | | // fmt.Println("File.Read err:",err) |
| | | // return nil,err |
| | | //} |
| | | |
| | | picMat, err := gocv.IMDecode(imgB, gocv.IMReadColor) |
| | | picMat, err := gocv.IMDecode(s.File, gocv.IMReadColor) |
| | | if err !=nil { |
| | | fmt.Println("gocv.IMDecode err:",err) |
| | | return nil,err |
| | |
| | | |
| | | fmt.Println("Len(sdkInfos)=",len(sdkInfos)) |
| | | for _,swt :=range sdkInfos{ |
| | | if swt.Sdktype =="FaceDetect"{ |
| | | fmt.Println("人脸检测结果长度:",len(swt.Sdkdata)) |
| | | if len(swt.Sdkdata)>1{ |
| | | var pfp protomsg.ParamFacePos |
| | | err := proto.Unmarshal(swt.Sdkdata, &pfp) |
| | | if err !=nil { |
| | | fmt.Println("faceDetect result unmarshal err:",err) |
| | | } else { |
| | | for _,face :=range pfp.Faces{ |
| | | fmt.Println("FacePos:",face.Pos) |
| | | fmt.Println("ThftResult:",face.Result) |
| | | } |
| | | } |
| | | } |
| | | } |
| | | //人脸检测的结果 |
| | | //if swt.Sdktype =="FaceDetect"{ |
| | | // fmt.Println("人脸检测结果长度:",len(swt.Sdkdata)) |
| | | // if len(swt.Sdkdata)>1{ |
| | | // var pfp protomsg.ParamFacePos |
| | | // err := proto.Unmarshal(swt.Sdkdata, &pfp) |
| | | // if err !=nil { |
| | | // fmt.Println("faceDetect result unmarshal err:",err) |
| | | // } else { |
| | | // for _,face :=range pfp.Faces{ |
| | | // fmt.Println("FacePos:",face.Pos) |
| | | // fmt.Println("ThftResult:",face.Result) |
| | | // } |
| | | // } |
| | | // } |
| | | //} |
| | | //人脸提取的结果 |
| | | if swt.Sdktype == "FaceExtract"{ |
| | | fmt.Println("人脸提取结果长度:",len(swt.Sdkdata)) |
| | | if len(swt.Sdkdata) > 1{ |
| | |
| | | if len(pff.ExtComp) >0 { |
| | | resultMap.Write(perId,pff.ExtComp) |
| | | |
| | | for _,fea :=range pff.ExtComp{ |
| | | base64Fea := base64.StdEncoding.EncodeToString(fea.Feats) |
| | | fmt.Println("perId:",perId) |
| | | fmt.Println("faceFeature:",base64Fea) |
| | | } |
| | | //for _,fea :=range pff.ExtComp{ |
| | | // base64Fea := base64.StdEncoding.EncodeToString(fea.Feats) |
| | | // fmt.Println("perId:",perId) |
| | | // fmt.Println("faceFeature:",base64Fea) |
| | | //} |
| | | } |
| | | } |
| | | break |