001/*
002 * Copyright (C) Photon Vision.
003 *
004 * This program is free software: you can redistribute it and/or modify
005 * it under the terms of the GNU General Public License as published by
006 * the Free Software Foundation, either version 3 of the License, or
007 * (at your option) any later version.
008 *
009 * This program is distributed in the hope that it will be useful,
010 * but WITHOUT ANY WARRANTY; without even the implied warranty of
011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
012 * GNU General Public License for more details.
013 *
014 * You should have received a copy of the GNU General Public License
015 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
016 */
017
018package org.photonvision.vision.pipeline;
019
020import org.photonvision.common.configuration.NeuralNetworkModelManager;
021import org.photonvision.vision.objects.Model;
022
023public class ObjectDetectionPipelineSettings extends AdvancedPipelineSettings {
024    public double confidence;
025    public double nms; // non maximal suppression
026    public String model;
027
028    public ObjectDetectionPipelineSettings() {
029        super();
030        this.pipelineType = PipelineType.ObjectDetection; // TODO: FIX this
031        this.outputShowMultipleTargets = true;
032        cameraExposureRaw = 20;
033        cameraAutoExposure = false;
034        ledMode = false;
035        confidence = .9;
036        nms = .45;
037        model =
038                NeuralNetworkModelManager.getInstance().getDefaultModel().map(Model::getName).orElse("");
039    }
040}