Reorganized code and added Morphology 2 cpp code

2011-06-21 05:06:25 +00:00
parent ae55138a49
commit 2819438fe6
11 changed files with 80 additions and 80 deletions
@@ -0,0 +1,52 @@
+/**
+ * @file AddingImages.cpp
+ * @brief Simple linear blender ( dst = alpha*src1 + beta*src2 )
+ * @author OpenCV team
+ */
+
+#include <cv.h>
+#include <highgui.h>
+#include <iostream>
+
+using namespace cv;
+
+/**
+ * @function main
+ * @brief Main function 
+ */
+int main( int argc, char** argv )
+{
+
+   double alpha = 0.5; double beta; double input; 
+
+   Mat src1, src2, dst;
+
+   /// Ask the user enter alpha
+   std::cout<<" Simple Linear Blender "<<std::endl;
+   std::cout<<"-----------------------"<<std::endl;
+   std::cout<<"* Enter alpha [0-1]: ";
+   std::cin>>input;
+
+   // We use the alpha provided by the user iff it is between 0 and 1
+   if( alpha >= 0 && alpha <= 1 )
+     { alpha = input; }
+
+   /// Read image ( same size, same type )
+   src1 = imread("../images/LinuxLogo.jpg");
+   src2 = imread("../images/WindowsLogo.jpg");
+
+   if( !src1.data ) { printf("Error loading src1 \n"); return -1; }
+   if( !src2.data ) { printf("Error loading src2 \n"); return -1; }
+
+   /// Create Windows
+   namedWindow("Linear Blend", 1);
+
+   beta = ( 1.0 - alpha );
+   addWeighted( src1, alpha, src2, beta, 0.0, dst);
+  
+   imshow( "Linear Blend", dst );
+
+
+   waitKey(0);
+   return 0;
+}
@@ -0,0 +1,58 @@
+/**
+ * @file BasicLinearTransforms.cpp
+ * @brief Simple program to change contrast and brightness
+ * @author OpenCV team
+ */
+
+#include <cv.h>
+#include <highgui.h>
+#include <iostream>
+
+using namespace cv;
+
+double alpha; /**< Simple contrast control */
+int beta;  /**< Simple brightness control */
+
+/**
+ * @function main
+ * @brief Main function 
+ */
+int main( int argc, char** argv )
+{
+   /// Read image given by user
+   Mat image = imread( argv[1] );
+   Mat new_image = Mat::zeros( image.size(), image.type() );
+
+   /// Initialize values 
+   std::cout<<" Basic Linear Transforms "<<std::endl;
+   std::cout<<"-------------------------"<<std::endl;
+   std::cout<<"* Enter the alpha value [1.0-3.0]: ";std::cin>>alpha;
+   std::cout<<"* Enter the beta value [0-100]: "; std::cin>>beta;
+
+
+   /// Do the operation new_image(i,j) = alpha*image(i,j) + beta
+   /// Instead of these 'for' loops we could have used simply:
+   /// image.convertTo(new_image, -1, alpha, beta);
+   /// but we wanted to show you how to access the pixels :)
+   for( int y = 0; y < image.rows; y++ )
+      { for( int x = 0; x < image.cols; x++ )
+           { for( int c = 0; c < 3; c++ )
+                {
+		  new_image.at<Vec3b>(y,x)[c] = saturate_cast<uchar>( alpha*( image.at<Vec3b>(y,x)[c] ) + beta );
+                }
+	   }
+      }
+
+   /// Create Windows
+   namedWindow("Original Image", 1);
+   namedWindow("New Image", 1);
+
+   /// Show stuff
+   imshow("Original Image", image);
+   imshow("New Image", new_image);
+
+   
+   /// Wait until user press some key
+   waitKey();
+   return 0;
+}
@@ -0,0 +1,105 @@
+/**
+ * @file Morphology_1.cpp
+ * @brief Erosion and Dilation sample code
+ * @author OpenCV team
+ */
+
+#include "opencv2/imgproc/imgproc.hpp"
+#include "opencv2/highgui/highgui.hpp"
+#include "highgui.h"
+#include <stdlib.h>
+#include <stdio.h>
+
+using namespace cv;
+
+/// Global variables
+Mat src, erosion_dst, dilation_dst;
+
+int erosion_elem = 0;
+int erosion_size = 0;
+int dilation_elem = 0;
+int dilation_size = 0;
+int const max_elem = 2;
+int const max_kernel_size = 21;
+
+/** Function Headers */
+void Erosion( int, void* );
+void Dilation( int, void* );
+
+/**
+ * @function main
+ */
+int main( int argc, char** argv )
+{
+  /// Load an image
+  src = imread( argv[1] );
+
+  if( !src.data )
+    { return -1; }
+    
+  /// Create windows
+  namedWindow( "Erosion Demo", CV_WINDOW_AUTOSIZE );
+  namedWindow( "Dilation Demo", CV_WINDOW_AUTOSIZE );
+  cvMoveWindow( "Dilation Demo", src.cols, 0 );
+
+  /// Create Erosion Trackbar
+  createTrackbar( "Element:\n 0: Rect \n 1: Cross \n 2: Ellipse", "Erosion Demo", 
+		  &erosion_elem, max_elem, 
+		  Erosion );
+
+  createTrackbar( "Kernel size:\n 2n +1", "Erosion Demo", 
+		  &erosion_size, max_kernel_size,
+		  Erosion );
+
+  /// Create Dilation Trackbar
+  createTrackbar( "Element:\n 0: Rect \n 1: Cross \n 2: Ellipse", "Dilation Demo", 
+		  &dilation_elem, max_elem, 
+		  Dilation );
+
+  createTrackbar( "Kernel size:\n 2n +1", "Dilation Demo", 
+		  &dilation_size, max_kernel_size,
+		  Dilation ); 
+
+  /// Default start
+  Erosion( 0, 0 );
+  Dilation( 0, 0 );
+
+  waitKey(0);
+  return 0;
+}
+
+/** 
+ * @function Erosion
+ */
+void Erosion( int, void* )
+{
+  int erosion_type;
+  if( erosion_elem == 0 ){ erosion_type = MORPH_RECT; }
+  else if( erosion_elem == 1 ){ erosion_type = MORPH_CROSS; }
+  else if( erosion_elem == 2) { erosion_type = MORPH_ELLIPSE; }
+
+  Mat element = getStructuringElement( erosion_type, 
+				       Size( 2*erosion_size + 1, 2*erosion_size+1 ),
+				       Point( erosion_size, erosion_size ) ); 
+  /// Apply the erosion operation
+  erode( src, erosion_dst, element );
+  imshow( "Erosion Demo", erosion_dst );  
+}
+
+/** 
+ * @function Dilation
+ */
+void Dilation( int, void* )
+{
+  int dilation_type;
+  if( dilation_elem == 0 ){ dilation_type = MORPH_RECT; }
+  else if( dilation_elem == 1 ){ dilation_type = MORPH_CROSS; }
+  else if( dilation_elem == 2) { dilation_type = MORPH_ELLIPSE; }
+
+  Mat element = getStructuringElement( dilation_type, 
+				       Size( 2*dilation_size + 1, 2*dilation_size+1 ),
+				       Point( dilation_size, dilation_size ) ); 
+  /// Apply the dilation operation
+  dilate( src, dilation_dst, element );
+  imshow( "Dilation Demo", dilation_dst );  
+}
@@ -0,0 +1,80 @@
+/**
+ * @file Morphology_2.cpp
+ * @brief Advanced morphology Transformations sample code
+ * @author OpenCV team
+ */
+
+#include "opencv2/imgproc/imgproc.hpp"
+#include "opencv2/highgui/highgui.hpp"
+#include <stdlib.h>
+#include <stdio.h>
+
+using namespace cv;
+
+/// Global variables
+Mat src, dst;
+
+int morph_elem = 0;
+int morph_size = 0;
+int morph_operator = 0;
+int const max_operator = 4;
+int const max_elem = 2;
+int const max_kernel_size = 21;
+
+char* window_name = "Morphology Transformations Demo";
+
+
+/** Function Headers */
+void Morphology_Operations( int, void* );
+
+/**
+ * @function main
+ */
+int main( int argc, char** argv )
+{
+  /// Load an image
+  src = imread( argv[1] );
+
+  if( !src.data )
+    { return -1; }
+    
+  /// Create window
+  namedWindow( window_name, CV_WINDOW_AUTOSIZE );
+
+  /// Create Trackbar to select Morphology operation
+  createTrackbar("Operator:\n 0: Opening - 1: Closing  \n 2: Gradient - 3: Top Hat \n 4: Black Hat", window_name, &morph_operator, max_operator, Morphology_Operations );
+
+  /// Create Trackbar to select kernel type
+  createTrackbar( "Element:\n 0: Rect - 1: Cross - 2: Ellipse", window_name, 
+		  &morph_elem, max_elem, 
+		  Morphology_Operations );
+
+  /// Create Trackbar to choose kernel size
+  createTrackbar( "Kernel size:\n 2n +1", window_name, 
+		  &morph_size, max_kernel_size,
+		  Morphology_Operations );
+
+  /// Default start
+  Morphology_Operations( 0, 0 );
+
+  waitKey(0);
+  return 0;
+}
+
+/** 
+ * @function Morphology_Operations
+ */
+void Morphology_Operations( int, void* )
+{
+
+  // Since MORPH_X : 2,3,4,5 and 6
+  int operation = morph_operator + 2;
+
+  Mat element = getStructuringElement( morph_elem, Size( 2*morph_size + 1, 2*morph_size+1 ), Point( morph_size, morph_size ) ); 
+
+  /// Apply the specified morphology operation
+  morphologyEx( src, dst, operation, element );
+  imshow( window_name, dst );  
+}
+
+
@@ -0,0 +1,109 @@
+/**
+ * file Smoothing.cpp
+ * brief Sample code for simple filters
+ * author OpenCV team
+ */
+#include <iostream>
+#include <vector>
+
+#include "opencv2/imgproc/imgproc.hpp"
+#include "opencv2/highgui/highgui.hpp"
+#include "opencv2/features2d/features2d.hpp"
+
+using namespace std;
+using namespace cv;
+
+/// Global Variables
+int DELAY_CAPTION = 1500;
+int DELAY_BLUR = 100;
+int MAX_KERNEL_LENGTH = 31;
+
+Mat src; Mat dst;
+char window_name[] = "Smoothing Demo";
+
+/// Function headers
+int display_caption( char* caption );
+int display_dst( int delay );
+
+
+/** 
+ * function main 
+ */
+int main( int argc, char** argv )
+{
+  namedWindow( window_name, CV_WINDOW_AUTOSIZE );
+
+  /// Load the source image
+  src = imread( "../images/lena.png", 1 ); 
+
+  if( display_caption( "Original Image" ) != 0 ) { return 0; }
+
+  dst = src.clone();
+  if( display_dst( DELAY_CAPTION ) != 0 ) { return 0; }
+
+
+  /// Applying Homogeneous blur 
+  if( display_caption( "Homogeneous Blur" ) != 0 ) { return 0; }
+
+  for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
+      { blur( src, dst, Size( i, i ), Point(-1,-1) );
+        if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
+
+
+  /// Applying Gaussian blur 
+  if( display_caption( "Gaussian Blur" ) != 0 ) { return 0; }
+
+  for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
+      { GaussianBlur( src, dst, Size( i, i ), 0, 0 );
+        if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
+
+
+  /// Applying Median blur
+  if( display_caption( "Median Blur" ) != 0 ) { return 0; }
+
+  for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
+      { medianBlur ( src, dst, i );
+        if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
+
+
+  /// Applying Bilateral Filter
+  if( display_caption( "Bilateral Blur" ) != 0 ) { return 0; }
+
+  for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
+      { bilateralFilter ( src, dst, i, i*2, i/2 );
+        if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
+
+  /// Wait until user press a key
+  display_caption( "End: Press a key!" );
+
+  waitKey(0);
+
+  return 0;
+}
+
+/**
+ * @function display_caption
+ */
+int display_caption( char* caption )
+{
+  dst = Mat::zeros( src.size(), src.type() );
+  putText( dst, caption, 
+           Point( src.cols/4, src.rows/2), 
+           CV_FONT_HERSHEY_COMPLEX, 1, Scalar(255, 255, 255) );
+
+  imshow( window_name, dst );
+  int c = waitKey( DELAY_CAPTION );
+  if( c >= 0 ) { return -1; }
+  return 0;
+}
+
+/**
+ * @function display_dst
+ */
+int display_dst( int delay )
+{
+  imshow( window_name, dst );
+  int c = waitKey ( delay );
+  if( c >= 0 ) { return -1; }
+  return 0;   
+}