示例#1
0
func main() {
	app = cli.NewApp()
	app.Name = "facecrop"
	app.Usage = "crop face from image by rect info arguments"

	app.Flags = []cli.Flag{
		cli.StringFlag{
			Name:  "path, p",
			Value: "",
			Usage: "path of image file.",
		},
		cli.StringFlag{
			Name:  "rect, r",
			Value: "",
			Usage: "rect to crop. (format: left,top,right,bottom)",
		},
		cli.StringFlag{
			Name:  "out, o",
			Value: "out",
			Usage: "path of output file.",
		},
	}

	app.Action = func(c *cli.Context) {
		pathArg := c.String("path")
		rectArg := c.String("rect")
		outArg := c.String("out")

		if len(pathArg) == 0 || len(rectArg) == 0 {
			println("path and rect are required.")
			cli.ShowAppHelp(c)
			return
		}

		fmt.Printf("Processing image file path: %s\n", pathArg)

		file_in, err := os.Open(pathArg)
		if err != nil {
			log.Fatal("File open error. err: ", err)
			os.Exit(255)
		}
		defer file_in.Close()

		image_in, err := jpeg.Decode(file_in)
		if err != nil {
			log.Fatal("Image decode error. err: ", err)
			os.Exit(255)
		}

		faceInfo, err := facecheck.ParseFaceInfo(rectArg)
		if err != nil {
			fmt.Printf("Parse Rect error. err: %v, rect: %s\n", err, rectArg)
		}

		faceInfo_json, _ := json.Marshal(faceInfo)
		fmt.Printf("Face info is: %s\n", faceInfo_json)

		writeImageFunc := func(r *facecheck.Rect, suffix string, i int) {
			fmt.Printf("%s rect is: %v\n", suffix, r)
			rect := image.Rect(r.Left, r.Top, r.Right, r.Bottom)
			file_out, err := os.Create(outArg + "_" + suffix + strconv.Itoa(i) + ".jpg")
			if err != nil {
				log.Fatal("File create error. err: ", err)
				os.Exit(255)
			}
			defer file_out.Close()
			croped := utils.CropImage(image_in, rect)
			jpeg.Encode(file_out, croped, &jpeg.Options{100})
		}

		for i, v := range faceInfo.Faces {
			writeImageFunc(&v, "Face", i)
		}
		for i, v := range faceInfo.Heads {
			writeImageFunc(&v, "Head", i)
		}

		fmt.Println("DONE!")
	}

	app.Run(os.Args)
}
示例#2
0
func procImage(img *model.ImageInfo, mongo *m.Mongo, p pool.Pool) {
	image_json, _ := json.Marshal(img)
	log.Printf("Processing image. image: %s", image_json)
	tcpConn, err := p.Get()
	if err != nil {
		log.Printf("Get connection from pool error. err: %v", err)
		return
	}
	defer tcpConn.Close()

	filePath := filepath.Join(config.ImageFilePath, img.File_path)
	log.Printf("Processing. path: %s", filePath)
	message := facecheck.BuildMessage(filePath)
	if message == nil {
		log.Printf("Build message Error.")
		return
	}

	writer := bufio.NewWriter(tcpConn)
	n, err := writer.Write(message)
	if err != nil {
		log.Printf("Write message to FaceRec server Error. err: = %v", err)
		return
	}
	if n != len(message) {
		log.Printf("Write message to FaceRec size Error. expect: %d, wrote: %d", len(message), n)
		return
	}
	err = writer.Flush()
	if err != nil {
		log.Printf("Flush message to FaceRec server Error. err: = %v", err)
		return
	}

	dataBuffer := bytes.Buffer{}
	b := make([]byte, 256)
	totalLen := int32(18)
	messageBody := ""
	isComplete := false
	for {
		n, err := tcpConn.Read(b)
		if err != nil {
			if err != io.EOF {
				log.Printf("Read message from FaceRec server Error. err: = %v", err)
			}
			return
		}
		dataBuffer.Write(b[:n])
		log.Printf("Read message from FaceRec server. n: %d, buffer: %x", n, dataBuffer.Bytes())
		if int32(dataBuffer.Len()) >= totalLen {
			totalLen, messageBody, isComplete = facecheck.ParseResult(dataBuffer.Bytes())
			log.Printf(`Read message from FaceRec server. totalLen: %d, messageBody: "%s", isComplete: %v`, totalLen, messageBody, isComplete)
			if isComplete {
				break
			}
		}
	}
	img.Has_face = (len(messageBody) != 0)
	img.Face_info = messageBody
	img.Face_checked = true

	if img.Has_face == false {
		log.Printf("NO FACE: %v", img)
		return
	} else {
		log.Printf("FOUND FACE: %v", img)
	}

	file_in, err := os.Open(filePath)
	if err != nil {
		log.Fatal("File open error. err: ", err)
		os.Exit(255)
	}
	defer file_in.Close()

	image_in, err := jpeg.Decode(file_in)
	if err != nil {
		log.Fatal("Image decode error. err: ", err)
		os.Exit(255)
	}
	file_out, err := os.Create("Original.jpg")
	if err != nil {
		log.Fatal("File create error. err: ", err)
		os.Exit(255)
	}
	defer file_out.Close()
	jpeg.Encode(file_out, image_in, &jpeg.Options{100})

	faceInfo, err := facecheck.ParseFaceInfo(img.Face_info)
	if err != nil {
		fmt.Printf("Parse Rect error. err: %v, rect: %s\n", err, img.Face_info)
	}
	faceInfo_json, _ := json.Marshal(faceInfo)
	fmt.Printf("Face info is: %s\n", faceInfo_json)

	writeImageFunc := func(r *facecheck.Rect, suffix string, i int) {
		fmt.Printf("%s rect is: %v\n", suffix, r)
		rect := image.Rect(r.Left, r.Top, r.Right, r.Bottom)
		file_out, err := os.Create("Out_" + suffix + strconv.Itoa(i) + ".jpg")
		if err != nil {
			log.Fatal("File create error. err: ", err)
			os.Exit(255)
		}
		defer file_out.Close()
		croped := utils.CropImage(image_in, rect)
		jpeg.Encode(file_out, croped, &jpeg.Options{100})
	}

	for i, v := range faceInfo.Faces {
		writeImageFunc(&v, "Face", i)
	}
	for i, v := range faceInfo.Heads {
		writeImageFunc(&v, "Head", i)
	}
}