OpenCV彩色影象轉為灰度


在前面的章節中,我們討論了如何讀取不同型別的輸入影象(二進位制,灰度,BGR等)。 在本章中,我們將學習如何將一種影象轉換為另一種影象。

org.opencv.imgproc包中名為Imgproc的類提供了將影象從一種顏色轉換為另一種顏色的方法。

將彩色影象轉換為灰度

使用cvtColor()方法將彩色影象轉換為灰度。 以下是此方法的語法。

cvtColor(Mat src, Mat dst, int code)

該方法接受以下引數 -

  • src - 表示來源的矩陣。
  • dst - 表示目的地的矩陣。
  • code - 表示轉換型別的整數程式碼,例如RGB到灰度。

通過將程式碼Imgproc.COLOR_RGB2GRAY與源矩陣和目標矩陣一起作為引數傳遞給cvtColor()方法,可以將彩色影象轉換為灰度。

範例

以下程式演示如何將彩色影象讀取為灰度影象並使用JavaFX視窗顯示。

package tw511.com;

import java.awt.image.BufferedImage;

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

import javafx.application.Application;
import javafx.embed.swing.SwingFXUtils;
import javafx.scene.Group;
import javafx.scene.Scene;
import javafx.scene.image.ImageView;
import javafx.scene.image.WritableImage;

import javafx.stage.Stage;

public class ColorToGrayscale extends Application {
   @Override
   public void start(Stage stage) throws Exception {
      WritableImage writableImage = loadAndConvert();

      // Setting the image view
      ImageView imageView = new ImageView(writableImage);

      // Setting the position of the image
      imageView.setX(10);
      imageView.setY(10);

      // setting the fit height and width of the image view
      imageView.setFitHeight(400);
      imageView.setFitWidth(650);

      // Setting the preserve ratio of the image view
      imageView.setPreserveRatio(true);

      // Creating a Group object
      Group root = new Group(imageView);

      // Creating a scene object
      Scene scene = new Scene(root, 650, 400);

      // Setting title to the Stage
      stage.setTitle("Colored to grayscale image");

      // Adding scene to the stage
      stage.setScene(scene);

      // Displaying the contents of the stage
      stage.show();
   } 
   public WritableImage loadAndConvert() throws Exception {
      //Loading the OpenCV core library
      System.loadLibrary( Core.NATIVE_LIBRARY_NAME );

      String input = "F:/worksp/opencv/images/sample.jpg";

      //Reading the image
      Mat src = Imgcodecs.imread(input);

      //Creating the empty destination matrix
      Mat dst = new Mat();

      //Converting the image to gray sacle and saving it in the dst matrix
      Imgproc.cvtColor(src, dst, Imgproc.COLOR_RGB2GRAY);

      //Extracting data from the transformed image (dst)
      byte[] data1 = new byte[dst.rows() * dst.cols() * (int)(dst.elemSize())];
      dst.get(0, 0, data1);

      //Creating Buffered image using the data
      BufferedImage bufImage = new BufferedImage(dst.cols(),dst.rows(), 
         BufferedImage.TYPE_BYTE_GRAY);

      //Setting the data elements to the image
      bufImage.getRaster().setDataElements(0, 0, dst.cols(), dst.rows(), data1);

      //Creating a WritableImage
      WritableImage writableImage = SwingFXUtils.toFXImage(bufImage, null);
      System.out.println("Converted to Grayscale");
      return writableImage;
   }
   public static void main(String args[]) throws Exception {
      launch(args);
   }
}

假定以下是上述程式中指定的輸入影象sample.jpg。如下圖所示 -

在執行程式時,將得到以下輸出。