Mercurial > hg > orthanc-java
annotate Samples/MammographyDeepLearning/src/main/java/ImageProcessing.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 |
rev | line source |
---|---|
28
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
1 /** |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
2 * SPDX-FileCopyrightText: 2023-2024 Sebastien Jodogne, UCLouvain, Belgium |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
3 * SPDX-License-Identifier: GPL-3.0-or-later |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
4 **/ |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
5 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
6 /** |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
7 * Java plugin for Orthanc |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
8 * Copyright (C) 2023-2024 Sebastien Jodogne, UCLouvain, Belgium |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
9 * |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
10 * This program is free software: you can redistribute it and/or |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
11 * modify it under the terms of the GNU General Public License as |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
12 * published by the Free Software Foundation, either version 3 of the |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
13 * License, or (at your option) any later version. |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
14 * |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
15 * This program is distributed in the hope that it will be useful, but |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
16 * WITHOUT ANY WARRANTY; without even the implied warranty of |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
18 * General Public License for more details. |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
19 * |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
20 * You should have received a copy of the GNU General Public License |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
21 * along with this program. If not, see <http://www.gnu.org/licenses/>. |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
22 **/ |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
23 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
24 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
25 import ai.djl.ndarray.NDArray; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
26 import ai.djl.ndarray.NDManager; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
27 import ai.djl.ndarray.types.DataType; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
28 import ai.djl.ndarray.types.Shape; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
29 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
30 import java.awt.*; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
31 import java.awt.image.BufferedImage; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
32 import java.awt.image.DataBuffer; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
33 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
34 public class ImageProcessing { |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
35 static BufferedImage resizeImage(BufferedImage originalImage, int targetWidth, int targetHeight) { |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
36 BufferedImage resizedImage = new BufferedImage(targetWidth, targetHeight, originalImage.getType()); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
37 Graphics2D graphics2D = resizedImage.createGraphics(); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
38 graphics2D.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
39 graphics2D.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
40 graphics2D.drawImage(originalImage, 0, 0, targetWidth, targetHeight, null); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
41 graphics2D.dispose(); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
42 return resizedImage; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
43 } |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
44 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
45 static NDArray imageToTensor(NDManager manager, |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
46 BufferedImage image) { |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
47 if (image.getType() != image.TYPE_USHORT_GRAY /* 16 bpp */ && |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
48 image.getType() != image.TYPE_BYTE_GRAY /* 8 bpp */) { |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
49 throw new IllegalArgumentException(); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
50 } |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
51 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
52 float pixels[] = new float[image.getHeight() * image.getWidth()]; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
53 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
54 DataBuffer db = image.getData().getDataBuffer(); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
55 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
56 int pos = 0; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
57 for (int y = 0; y < image.getHeight(); y++) { |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
58 for (int x = 0; x < image.getWidth(); x++, pos++) { |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
59 pixels[pos] = db.getElemFloat(pos); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
60 } |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
61 } |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
62 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
63 return manager.create(pixels, new Shape(1, image.getHeight(), image.getWidth())); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
64 } |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
65 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
66 static NDArray standardize(NDArray image) { |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
67 if (image.getDataType() != DataType.FLOAT32 || |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
68 image.getShape().dimension() != 3 || |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
69 (image.getShape().get(0) != 1 && |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
70 image.getShape().get(0) != 3)) { |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
71 throw new IllegalArgumentException(); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
72 } |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
73 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
74 // Standardize the image to zero mean and 1 standard deviation |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
75 NDArray doubleImage = image.toType(DataType.FLOAT64, false); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
76 NDArray squared = doubleImage.mul(doubleImage); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
77 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
78 double asum = doubleImage.sum().getDouble(); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
79 double asumOfSquares = squared.sum().getDouble(); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
80 double n = doubleImage.getShape().size(); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
81 double amean = asum / n; |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
82 double astd = Math.sqrt((asumOfSquares - asum * asum / n) / n); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
83 |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
84 return image.add(-amean).div(astd); |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
85 } |
43923934e934
added sample: deep learning for mammography
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
86 } |