LoginSignup
0
0

More than 3 years have passed since last update.

Desktop : OpenCV Customized Filter

Posted at
Goal

OpenCV Customized Filter.

import org.opencv.core.Core;
import org.opencv.core.CvType;
import org.opencv.core.Mat;
import org.opencv.imgcodecs.Imgcodecs;
import org.opencv.imgproc.Imgproc;

import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.image.BufferedImage;

public class OpenCV_Customized {

    static { System.loadLibrary(Core.NATIVE_LIBRARY_NAME);}
    private JFrame frmjavaSwing;

    /**
        * Launch the application.
        */
    public static void main(String[] args) {
        EventQueue.invokeLater(new Runnable() {
            @Override
            public void run() {
                try {
                    OpenCV_Customized window = new OpenCV_Customized();
                    window.frmjavaSwing.setVisible(true);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
    }

    /**
        * Create the application.
        */
    public OpenCV_Customized(){
        init();
    }

    /**
        * Init tha contents of the frame.
        */
        private void init(){
            final Mat src = Imgcodecs.imread(
                    "D:\\projects\\Java\\OpenCV_Samples\\resource\\imgs\\lena.jpg");

            BufferedImage image=matToBufferedImage(src);

            frmjavaSwing = new JFrame();
            frmjavaSwing.setTitle("opencv ¦Û¤vªºfilter½m²ß");
            frmjavaSwing.setBounds(100, 100, 560, 620);
            frmjavaSwing.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
            frmjavaSwing.getContentPane().setLayout(null);

            final JLabel lblNewLabel = new JLabel("");
            lblNewLabel.setBounds(5, 60, image.getHeight()+10, image.getWidth()+10);
            lblNewLabel.setIcon(new ImageIcon(image));
            frmjavaSwing.getContentPane().add(lblNewLabel);

            JButton btnR0 = new JButton("C0");
            btnR0.addActionListener(new ActionListener() {
                public void actionPerformed(ActionEvent arg0) {
                    BufferedImage newImage=matToBufferedImage(Convolution(src,0));
                    lblNewLabel.setIcon(new ImageIcon(newImage));
                }
            });
            btnR0.setBounds(5, 10, 63, 23);
            frmjavaSwing.getContentPane().add(btnR0);

            JButton btnR1 = new JButton("C1");
            btnR1.addActionListener(new ActionListener() {
                public void actionPerformed(ActionEvent arg0) {
                    BufferedImage newImage=matToBufferedImage(Convolution(src,1));
                    lblNewLabel.setIcon(new ImageIcon(newImage));
                }
            });
            btnR1.setBounds(72, 10, 63, 23);
            frmjavaSwing.getContentPane().add(btnR1);

            JButton btnR2 = new JButton("C2");
            btnR2.addActionListener(new ActionListener() {
                public void actionPerformed(ActionEvent arg0) {
                    BufferedImage newImage=matToBufferedImage(Convolution(src,2));
                    lblNewLabel.setIcon(new ImageIcon(newImage));
                }
            });
            btnR2.setBounds(136, 10, 63, 23);
            frmjavaSwing.getContentPane().add(btnR2);

            JButton btnR3 = new JButton("C3");
            btnR3.addActionListener(new ActionListener() {
                public void actionPerformed(ActionEvent arg0) {
                    BufferedImage newImage=matToBufferedImage(Convolution(src,3));
                    lblNewLabel.setIcon(new ImageIcon(newImage));
                }
            });
            btnR3.setBounds(202, 10, 63, 23);
            frmjavaSwing.getContentPane().add(btnR3);

            JButton btnR4 = new JButton("C4");
            btnR4.addActionListener(new ActionListener() {
                public void actionPerformed(ActionEvent arg0) {
                    BufferedImage newImage=matToBufferedImage(Convolution(src,4));
                    lblNewLabel.setIcon(new ImageIcon(newImage));
                }
            });
            btnR4.setBounds(268, 10, 63, 23);
            frmjavaSwing.getContentPane().add(btnR4);

            JButton btnR5 = new JButton("C5");
            btnR5.addActionListener(new ActionListener() {
                public void actionPerformed(ActionEvent arg0) {
                    BufferedImage newImage=matToBufferedImage(Convolution(src,5));
                    lblNewLabel.setIcon(new ImageIcon(newImage));
                }
            });
            btnR5.setBounds(336, 10, 63, 23);
            frmjavaSwing.getContentPane().add(btnR5);
        }

    public Mat Convolution(Mat source,int type){
        Mat kernel =new Mat(3,3, CvType.CV_32F);
        if (type==0){
            kernel.put(0, 0, new float[]{0,-1,5});
            kernel.put(1, 0, new float[]{-1,4,-1});
            kernel.put(2, 0, new float[]{0,-1,3});

        }else if (type==1){
            kernel.put(0, 0, new float[]{0,-1,5});
            kernel.put(1, 0, new float[]{-1,4,-1});
            kernel.put(2, 0, new float[]{0,-1,6});


        }else if (type==2){
            kernel.put(0, 0, new float[]{0,-1,5});
            kernel.put(1, 0, new float[]{-1,4,-1});
            kernel.put(2, 0, new float[]{0,-1,9});


        }else if (type==3){
            kernel.put(0, 0, new float[]{-1,-2,1});
            kernel.put(1, 0, new float[]{0,0,0});
            kernel.put(2, 0, new float[]{1,2,1});



        }else if (type==4){
            kernel.put(0, 0, new float[]{-3,-0,-3});
            kernel.put(1, 0, new float[]{-10,0,10});
            kernel.put(2, 0, new float[]{-3,10,3});



        }else if (type==5){
            kernel.put(0, 0, new float[]{-3,-0,3});
            kernel.put(1, 0, new float[]{-10,0,10});
            kernel.put(2, 0, new float[]{-3,10,3});

        }
        Mat destination=new Mat(source.rows(),source.cols(),source.type());
        Imgproc.filter2D(source, destination, -1, kernel);
        return destination;

    }

    public BufferedImage matToBufferedImage(Mat matrix) {
        int cols = matrix.cols();
        int rows = matrix.rows();
        int elemSize = (int)matrix.elemSize();
        byte[] data = new byte[cols * rows * elemSize];
        int type;
        matrix.get(0, 0, data);
        switch (matrix.channels()) {
            case 1:
                type = BufferedImage.TYPE_BYTE_GRAY;
                break;
            case 3:
                type = BufferedImage.TYPE_3BYTE_BGR;
                // bgr to rgb
                byte b;
                for(int i=0; i<data.length; i=i+3) {
                    b = data[i];
                    data[i] = data[i+2];
                    data[i+2] = b;
                }
                break;
            default:
                return null;
        }
        BufferedImage image2 = new BufferedImage(cols, rows, type);
        image2.getRaster().setDataElements(0, 0, cols, rows, data);
        return image2;
    }

}
Result

opencv_customized.JPG

0
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
0