mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-07 17:44:04 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use directly HighGui.toBufferedImage().

Browse Source
This commit is contained in:
catree 2018-05-25 16:23:42 +02:00
parent 0a6d190095
commit adef31097d
3 changed files with 67 additions and 109 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
samples/java/tutorial_code/ImgProc/erosion_dilatation/MorphologyDemo1.java
View File

@ -1,9 +1,8 @@
import java.awt.BorderLayout; import java.awt.BorderLayout;
import java.awt.Container; import java.awt.Container;
import java.awt.Image;
import java.awt.event.ActionEvent; import java.awt.event.ActionEvent;
import java.awt.event.ActionListener; import java.awt.event.ActionListener;
import java.awt.image.BufferedImage;
import java.awt.image.DataBufferByte;
import javax.swing.BoxLayout; import javax.swing.BoxLayout;
import javax.swing.ImageIcon; import javax.swing.ImageIcon;
@ -19,6 +18,7 @@ import org.opencv.core.Core;
import org.opencv.core.Mat; import org.opencv.core.Mat;
import org.opencv.core.Point; import org.opencv.core.Point;
import org.opencv.core.Size; import org.opencv.core.Size;
import org.opencv.highgui.HighGui;
import org.opencv.imgcodecs.Imgcodecs; import org.opencv.imgcodecs.Imgcodecs;
import org.opencv.imgproc.Imgproc; import org.opencv.imgproc.Imgproc;
@ -46,7 +46,7 @@ public class MorphologyDemo1 {
frame = new JFrame("Erosion and dilatation demo"); frame = new JFrame("Erosion and dilatation demo");
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
// Set up the content pane. // Set up the content pane.
BufferedImage img = toBufferedImage(matImgSrc); Image img = HighGui.toBufferedImage(matImgSrc);
addComponentsToPane(frame.getContentPane(), img); addComponentsToPane(frame.getContentPane(), img);
// Use the content pane's default BorderLayout. No need for // Use the content pane's default BorderLayout. No need for
// setLayout(new BorderLayout()); // setLayout(new BorderLayout());
@ -55,7 +55,7 @@ public class MorphologyDemo1 {
frame.setVisible(true); frame.setVisible(true);
} }
private void addComponentsToPane(Container pane, BufferedImage img) { private void addComponentsToPane(Container pane, Image img) {
if (!(pane.getLayout() instanceof BorderLayout)) { if (!(pane.getLayout() instanceof BorderLayout)) {
pane.add(new JLabel("Container doesn't use BorderLayout!")); pane.add(new JLabel("Container doesn't use BorderLayout!"));
return; return;
@ -115,20 +115,6 @@ public class MorphologyDemo1 {
pane.add(imgLabel, BorderLayout.CENTER); pane.add(imgLabel, BorderLayout.CENTER);
} }
private BufferedImage toBufferedImage(Mat matrix) {
int type = BufferedImage.TYPE_BYTE_GRAY;
if (matrix.channels() > 1) {
type = BufferedImage.TYPE_3BYTE_BGR;
}
int bufferSize = matrix.channels() * matrix.cols() * matrix.rows();
byte[] buffer = new byte[bufferSize];
matrix.get(0, 0, buffer); // get all the pixels
BufferedImage image = new BufferedImage(matrix.cols(), matrix.rows(), type);
final byte[] targetPixels = ((DataBufferByte) image.getRaster().getDataBuffer()).getData();
System.arraycopy(buffer, 0, targetPixels, 0, buffer.length);
return image;
}
private void update() { private void update() {
Mat element = Imgproc.getStructuringElement(elementType, new Size(2 * kernelSize + 1, 2 * kernelSize + 1), Mat element = Imgproc.getStructuringElement(elementType, new Size(2 * kernelSize + 1, 2 * kernelSize + 1),
new Point(kernelSize, kernelSize)); new Point(kernelSize, kernelSize));
@ -138,7 +124,7 @@ public class MorphologyDemo1 {
} else { } else {
Imgproc.dilate(matImgSrc, matImgDst, element); Imgproc.dilate(matImgSrc, matImgDst, element);
} }
BufferedImage img = toBufferedImage(matImgDst); Image img = HighGui.toBufferedImage(matImgDst);
imgLabel.setIcon(new ImageIcon(img)); imgLabel.setIcon(new ImageIcon(img));
frame.repaint(); frame.repaint();
} }

24
samples/java/tutorial_code/ImgProc/opening_closing_hats/MorphologyDemo2.java
View File

@ -1,9 +1,8 @@
import java.awt.BorderLayout; import java.awt.BorderLayout;
import java.awt.Container; import java.awt.Container;
import java.awt.Image;
import java.awt.event.ActionEvent; import java.awt.event.ActionEvent;
import java.awt.event.ActionListener; import java.awt.event.ActionListener;
import java.awt.image.BufferedImage;
import java.awt.image.DataBufferByte;
import javax.swing.BoxLayout; import javax.swing.BoxLayout;
import javax.swing.ImageIcon; import javax.swing.ImageIcon;
@ -19,6 +18,7 @@ import org.opencv.core.Core;
import org.opencv.core.Mat; import org.opencv.core.Mat;
import org.opencv.core.Point; import org.opencv.core.Point;
import org.opencv.core.Size; import org.opencv.core.Size;
import org.opencv.highgui.HighGui;
import org.opencv.imgcodecs.Imgcodecs; import org.opencv.imgcodecs.Imgcodecs;
import org.opencv.imgproc.Imgproc; import org.opencv.imgproc.Imgproc;
@ -48,7 +48,7 @@ public class MorphologyDemo2 {
frame = new JFrame("Morphology Transformations demo"); frame = new JFrame("Morphology Transformations demo");
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
// Set up the content pane. // Set up the content pane.
BufferedImage img = toBufferedImage(matImgSrc); Image img = HighGui.toBufferedImage(matImgSrc);
addComponentsToPane(frame.getContentPane(), img); addComponentsToPane(frame.getContentPane(), img);
// Use the content pane's default BorderLayout. No need for // Use the content pane's default BorderLayout. No need for
// setLayout(new BorderLayout()); // setLayout(new BorderLayout());
@ -57,7 +57,7 @@ public class MorphologyDemo2 {
frame.setVisible(true); frame.setVisible(true);
} }
private void addComponentsToPane(Container pane, BufferedImage img) { private void addComponentsToPane(Container pane, Image img) {
if (!(pane.getLayout() instanceof BorderLayout)) { if (!(pane.getLayout() instanceof BorderLayout)) {
pane.add(new JLabel("Container doesn't use BorderLayout!")); pane.add(new JLabel("Container doesn't use BorderLayout!"));
return; return;
@ -117,26 +117,12 @@ public class MorphologyDemo2 {
pane.add(imgLabel, BorderLayout.CENTER); pane.add(imgLabel, BorderLayout.CENTER);
} }
private BufferedImage toBufferedImage(Mat matrix) {
int type = BufferedImage.TYPE_BYTE_GRAY;
if (matrix.channels() > 1) {
type = BufferedImage.TYPE_3BYTE_BGR;
}
int bufferSize = matrix.channels() * matrix.cols() * matrix.rows();
byte[] buffer = new byte[bufferSize];
matrix.get(0, 0, buffer); // get all the pixels
BufferedImage image = new BufferedImage(matrix.cols(), matrix.rows(), type);
final byte[] targetPixels = ((DataBufferByte) image.getRaster().getDataBuffer()).getData();
System.arraycopy(buffer, 0, targetPixels, 0, buffer.length);
return image;
}
private void update() { private void update() {
Mat element = Imgproc.getStructuringElement(elementType, new Size(2 * kernelSize + 1, 2 * kernelSize + 1), Mat element = Imgproc.getStructuringElement(elementType, new Size(2 * kernelSize + 1, 2 * kernelSize + 1),
new Point(kernelSize, kernelSize)); new Point(kernelSize, kernelSize));
Imgproc.morphologyEx(matImgSrc, matImgDst, morphOpType, element); Imgproc.morphologyEx(matImgSrc, matImgDst, morphOpType, element);
BufferedImage img = toBufferedImage(matImgDst); Image img = HighGui.toBufferedImage(matImgDst);
imgLabel.setIcon(new ImageIcon(img)); imgLabel.setIcon(new ImageIcon(img));
frame.repaint(); frame.repaint();
} }

92
samples/java/tutorial_code/ImgProc/tutorial_template_matching/MatchTemplateDemo.java
View File

@ -1,16 +1,22 @@
import org.opencv.core.*; import java.awt.GridLayout;
import org.opencv.core.Point; import java.awt.Image;
import org.opencv.imgcodecs.Imgcodecs; import java.util.Hashtable;
import org.opencv.imgproc.Imgproc;
import javax.swing.*; import javax.swing.ImageIcon;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JSlider;
import javax.swing.event.ChangeEvent; import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener; import javax.swing.event.ChangeListener;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.awt.image.DataBufferByte;
import java.util.*; import org.opencv.core.Core;
import org.opencv.core.CvType;
import org.opencv.core.Mat;
import org.opencv.core.Point;
import org.opencv.core.Scalar;
import org.opencv.highgui.HighGui;
import org.opencv.imgcodecs.Imgcodecs;
import org.opencv.imgproc.Imgproc;
class MatchTemplateDemoRun implements ChangeListener { class MatchTemplateDemoRun implements ChangeListener {
@ -26,9 +32,7 @@ class MatchTemplateDemoRun implements ChangeListener{
//! [declare] //! [declare]
public void run(String[] args) { public void run(String[] args) {
if (args.length < 2) {
if (args.length < 2)
{
System.out.println("Not enough parameters"); System.out.println("Not enough parameters");
System.out.println("Program arguments:\n<image_name> <template_name> [<mask_name>]"); System.out.println("Program arguments:\n<image_name> <template_name> [<mask_name>]");
System.exit(-1); System.exit(-1);
@ -45,19 +49,16 @@ class MatchTemplateDemoRun implements ChangeListener{
mask = Imgcodecs.imread(args[2], Imgcodecs.IMREAD_COLOR); mask = Imgcodecs.imread(args[2], Imgcodecs.IMREAD_COLOR);
} }
if(img.empty() || templ.empty() || (use_mask && mask.empty())) if (img.empty() || templ.empty() || (use_mask && mask.empty())) {
{
System.out.println("Can't read one of the images"); System.out.println("Can't read one of the images");
System.exit(-1); System.exit(-1);
} }
matchingMethod(); matchingMethod();
createJFrame(); createJFrame();
} }
private void matchingMethod() { private void matchingMethod() {
Mat result = new Mat(); Mat result = new Mat();
//! [copy_source] //! [copy_source]
@ -76,12 +77,12 @@ class MatchTemplateDemoRun implements ChangeListener{
//! [match_template] //! [match_template]
/// Do the Matching and Normalize /// Do the Matching and Normalize
Boolean method_accepts_mask = (Imgproc.TM_SQDIFF == match_method || Boolean method_accepts_mask = (Imgproc.TM_SQDIFF == match_method || match_method == Imgproc.TM_CCORR_NORMED);
match_method == Imgproc.TM_CCORR_NORMED); if (use_mask && method_accepts_mask) {
if (use_mask && method_accepts_mask) Imgproc.matchTemplate(img, templ, result, match_method, mask);
{ Imgproc.matchTemplate( img, templ, result, match_method, mask); } } else {
else Imgproc.matchTemplate(img, templ, result, match_method);
{ Imgproc.matchTemplate( img, templ, result, match_method); } }
//! [match_template] //! [match_template]
//! [normalize] //! [normalize]
@ -90,7 +91,6 @@ class MatchTemplateDemoRun implements ChangeListener{
//! [best_match] //! [best_match]
/// Localizing the best match with minMaxLoc /// Localizing the best match with minMaxLoc
double minVal; double maxVal;
Point matchLoc; Point matchLoc;
Core.MinMaxLocResult mmr = Core.minMaxLoc(result); Core.MinMaxLocResult mmr = Core.minMaxLoc(result);
@ -98,55 +98,42 @@ class MatchTemplateDemoRun implements ChangeListener{
//! [match_loc] //! [match_loc]
/// For SQDIFF and SQDIFF_NORMED, the best matches are lower values. /// For SQDIFF and SQDIFF_NORMED, the best matches are lower values.
// For all the other methods, the higher the better /// For all the other methods, the higher the better
if( match_method == Imgproc.TM_SQDIFF || match_method == Imgproc.TM_SQDIFF_NORMED ) if (match_method == Imgproc.TM_SQDIFF || match_method == Imgproc.TM_SQDIFF_NORMED) {
{ matchLoc = mmr.minLoc; } matchLoc = mmr.minLoc;
else } else {
{ matchLoc = mmr.maxLoc; } matchLoc = mmr.maxLoc;
}
//! [match_loc] //! [match_loc]
//! [imshow] //! [imshow]
/// Show me what you got /// Show me what you got
Imgproc.rectangle(img_display, matchLoc, new Point(matchLoc.x + templ.cols(), Imgproc.rectangle(img_display, matchLoc, new Point(matchLoc.x + templ.cols(), matchLoc.y + templ.rows()),
matchLoc.y + templ.rows()), new Scalar(0, 0, 0), 2, 8, 0); new Scalar(0, 0, 0), 2, 8, 0);
Imgproc.rectangle(result, matchLoc, new Point(matchLoc.x + templ.cols(), Imgproc.rectangle(result, matchLoc, new Point(matchLoc.x + templ.cols(), matchLoc.y + templ.rows()),
matchLoc.y + templ.rows()), new Scalar(0, 0, 0), 2, 8, 0); new Scalar(0, 0, 0), 2, 8, 0);
Image tmpImg = toBufferedImage(img_display); Image tmpImg = HighGui.toBufferedImage(img_display);
ImageIcon icon = new ImageIcon(tmpImg); ImageIcon icon = new ImageIcon(tmpImg);
imgDisplay.setIcon(icon); imgDisplay.setIcon(icon);
result.convertTo(result, CvType.CV_8UC1, 255.0); result.convertTo(result, CvType.CV_8UC1, 255.0);
tmpImg = toBufferedImage(result); tmpImg = HighGui.toBufferedImage(result);
icon = new ImageIcon(tmpImg); icon = new ImageIcon(tmpImg);
resultDisplay.setIcon(icon); resultDisplay.setIcon(icon);
//! [imshow] //! [imshow]
} }
@Override
public void stateChanged(ChangeEvent e) { public void stateChanged(ChangeEvent e) {
JSlider source = (JSlider) e.getSource(); JSlider source = (JSlider) e.getSource();
if (!source.getValueIsAdjusting()) { if (!source.getValueIsAdjusting()) {
match_method = (int)source.getValue(); match_method = source.getValue();
matchingMethod(); matchingMethod();
} }
} }
public Image toBufferedImage(Mat m) {
int type = BufferedImage.TYPE_BYTE_GRAY;
if ( m.channels() > 1 ) {
type = BufferedImage.TYPE_3BYTE_BGR;
}
int bufferSize = m.channels()*m.cols()*m.rows();
byte [] b = new byte[bufferSize];
m.get(0,0,b); // get all the pixels
BufferedImage image = new BufferedImage(m.cols(),m.rows(), type);
final byte[] targetPixels = ((DataBufferByte) image.getRaster().getDataBuffer()).getData();
System.arraycopy(b, 0, targetPixels, 0, b.length);
return image;
}
private void createJFrame() { private void createJFrame() {
String title = "Source image; Control; Result image"; String title = "Source image; Control; Result image";
JFrame frame = new JFrame(title); JFrame frame = new JFrame(title);
frame.setLayout(new GridLayout(2, 2)); frame.setLayout(new GridLayout(2, 2));
@ -164,7 +151,7 @@ class MatchTemplateDemoRun implements ChangeListener{
slider.setMinorTickSpacing(1); slider.setMinorTickSpacing(1);
// Customizing the labels // Customizing the labels
Hashtable labelTable = new Hashtable(); Hashtable<Integer, JLabel> labelTable = new Hashtable<>();
labelTable.put(new Integer(0), new JLabel("0 - SQDIFF")); labelTable.put(new Integer(0), new JLabel("0 - SQDIFF"));
labelTable.put(new Integer(1), new JLabel("1 - SQDIFF NORMED")); labelTable.put(new Integer(1), new JLabel("1 - SQDIFF NORMED"));
labelTable.put(new Integer(2), new JLabel("2 - TM CCORR")); labelTable.put(new Integer(2), new JLabel("2 - TM CCORR"));
@ -184,8 +171,7 @@ class MatchTemplateDemoRun implements ChangeListener{
} }
} }
public class MatchTemplateDemo public class MatchTemplateDemo {
{
public static void main(String[] args) { public static void main(String[] args) {
// load the native OpenCV library // load the native OpenCV library
System.loadLibrary(Core.NATIVE_LIBRARY_NAME); System.loadLibrary(Core.NATIVE_LIBRARY_NAME);