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);
+    }
+}