Mercurial > hg > orthanc-java
diff Samples/MammographyDeepLearning/src/main/java/Detection.java @ 28:43923934e934
added sample: deep learning for mammography
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Wed, 12 Jun 2024 13:58:29 +0200 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Samples/MammographyDeepLearning/src/main/java/Detection.java Wed Jun 12 13:58:29 2024 +0200 @@ -0,0 +1,56 @@ +/** + * SPDX-FileCopyrightText: 2023-2024 Sebastien Jodogne, UCLouvain, Belgium + * SPDX-License-Identifier: GPL-3.0-or-later + **/ + +/** + * Java plugin for Orthanc + * Copyright (C) 2023-2024 Sebastien Jodogne, UCLouvain, Belgium + * + * This program is free software: you can redistribute it and/or + * modify it under the terms of the GNU General Public License as + * published by the Free Software Foundation, either version 3 of the + * License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + **/ + + +import java.io.Serializable; + +class Detection implements Comparable<Detection>, Serializable { + private Rectangle rectangle; + private int label; + private double score; + + public Detection(Rectangle rectangle, + int label, + double score) { + this.rectangle = rectangle; + this.label = label; + this.score = score; + } + + public Rectangle getRectangle() { + return rectangle; + } + + public int getLabel() { + return label; + } + + public double getScore() { + return score; + } + + @Override + public int compareTo(Detection peak) { + return Double.compare(peak.score, score); + } +}