Chiebot-Mirror
/
opencv
8
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use directly HighGui.toBufferedImage().

Browse Source
pull/11592/head
catree 7 years ago
parent 0a6d190095
commit adef31097d
3 changed files with 67 additions and 109 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 24
      samples/java/tutorial_code/ImgProc/erosion_dilatation/MorphologyDemo1.java
  2. 24
      samples/java/tutorial_code/ImgProc/opening_closing_hats/MorphologyDemo2.java
  3. 94
      samples/java/tutorial_code/ImgProc/tutorial_template_matching/MatchTemplateDemo.java

24
samples/java/tutorial_code/ImgProc/erosion_dilatation/MorphologyDemo1.java
Unescape 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
Unescape 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();
} }

94
samples/java/tutorial_code/ImgProc/tutorial_template_matching/MatchTemplateDemo.java
Unescape 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.ImageIcon;
import javax.swing.*; 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);

Write Preview
Loading…
Cancel
Save