one.empty3.feature.FaceDetect Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2022-2023. Manuel Daniel Dahmen
 *
 *
 *    Copyright 2012-2023 Manuel Daniel Dahmen
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

package one.empty3.feature;

import one.empty3.feature.jviolajones.Detector;
import one.empty3.io.ProcessFile;

import java.awt.*;
import java.io.*;
import java.util.List;

public class FaceDetect extends ProcessFile {
    public boolean process(File in, File out) {

        File img = in;
        InputStream haarXml = Detector.class.getResourceAsStream("/haarcascade_frontalface_alt2.xml");

        Detector detector = new Detector(haarXml);
        List res = null;
        try {
            File file = new File(getOutputDirectory()
                    .getAbsolutePath() + File.separator + "faces.csv");
            PrintWriter printWriter = new PrintWriter(new FileOutputStream(file));
            res = detector.getFaces(img.getAbsolutePath(), 1, 1.25f, 0.1f, 1, true);
            if (res.size() == 0) {
                System.out.printf("No face\n");
            } else {
                for (Rectangle re : res) {
                    System.out.printf("Face %d\n%d\n%d\n%d\n",
                            re.x, re.y, re.width, re.height);
                    printWriter.println(String.format("%s, %d, %d, %d, %d", out.getName(), re.x, re.y, re.width, re.height));
                }
            }
            printWriter.close();

            addSource(file);

            return res.size() > 0;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return false;
    }

}