Changeset 33170 for other-projects/is-sheet-music-encore/trunk/image-identification-terminal/javaImageClassifier.java

Timestamp:

2019-06-20T20:42:56+12:00 (5 years ago)

Author:

cpb16

Message:

refined houghlineP alogirthm

File:

• other-projects/is-sheet-music-encore/trunk/image-identification-terminal/javaImageClassifier.java (modified) (6 diffs)

other-projects/is-sheet-music-encore/trunk/image-identification-terminal/javaImageClassifier.java

@@ -33 +33 @@
     public static void main(String[] args) {    
         try {
-            if (args.length != 2) {
-            System.out.println("Usage: imageClassifier <inputFilename> <classifierType>");
+            if (args.length != 3) {
+            System.out.println("Usage: imageClassifier <inputFilename> <classifierType> <outputFilename>");
             }
             else {
                 Boolean result = null;
                 String imageFilename = args[0];
-                String classifierType = args[1];            
+                String classifierType = args[1];
+                String outputFilename = args[2];            
                 //Execute classifierType defined from arguement
                 switch(classifierType){
                 case "houghlinesP":
-                    result = setup_HoughLineP(imageFilename); //true or false
+                    result = setup_HoughLinesP(imageFilename); //true or false
+                    break;
+                case "houghlinesP-refined":
+                    result = setup_HoughLinesP_refined(imageFilename);
                     break;
                 default:
-                    System.out.println("unknown");
+                    System.out.println("unknown algorithm");
                     break;
                 }           
                 //Write output to disc
-                File log = new File("log.txt");
+                File log = new File(outputFilename);
                 FileWriter fileWriter = new FileWriter(log, true);
                 BufferedWriter bw = new BufferedWriter(fileWriter);
@@ -66 +70 @@
     //True = 1 + Filename + Status
     //False= 0 + Filename + Status
-    private static Boolean setup_HoughLineP(String filename){
+    
+ //******************
+ //CLASSIFIER FUNCTIONS
+ //******************
+    private static Boolean setup_HoughLinesP(String filename){
     System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
     Boolean isSheetMusic = null;
@@ -85 +93 @@
         // Draw the lines
         for (int x = 0; x < linesP.rows(); x++) {
-        double[] l = linesP.get(x, 0);
-        Imgproc.line(edgesDetectedRGB, new Point(l[0], l[1]), new Point(l[2], l[3]), new Scalar(0, 0, 255), 3, Imgproc.LINE_AA, 0);
+            double[] l = linesP.get(x, 0);
+            Imgproc.line(edgesDetectedRGB, new Point(l[0], l[1]), new Point(l[2], l[3]), new Scalar(0, 0, 255), 3, Imgproc.LINE_AA, 0);
         }
         
@@ -92 +100 @@
         BufferedImage toBeClassifiedImg = toBufferedImage(edgesDetectedRGB);
         //Calculate if its sheet music or not
-        isSheetMusic = classifier_HoughLineP(toBeClassifiedImg);  
-                
-        //Save Processed Image
-        String processedFile = filename;
-        if (isSheetMusic == true) {
-        processedFile = "proc_T_"+filename;
-        }else {
-        processedFile = "proc_F_"+filename;
-        }
-        imwrite(processedFile, edgesDetectedRGB);                   
+        isSheetMusic = classifier_HoughLinesP(toBeClassifiedImg);               
     }
     catch(Exception e){
@@ -107 +106 @@
         }
         return isSheetMusic;
-    /*    
-    if (isSheetMusic == true){    
-        return (1 + "\t" + "Filename: " + filename + "  Status: " + isSheetMusic  +"\t" );
-    }   
-    else{
-        return (0 + "\t" + "Filename: " + filename + "  Status: " + isSheetMusic  +"\t" );
-    } */
  }
  
+     private static Boolean setup_HoughLinesP_refined(String filename){
+    System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
+    Boolean isSheetMusic = null;
+    try{
+        //Variables
+        Mat edgesDetected = new Mat();
+        Mat edgesDetectedRGB = new Mat();
+        Mat edgesDetectedRGBProb;
+        // Load an image
+        Mat original = Imgcodecs.imread(filename, Imgcodecs.IMREAD_GRAYSCALE);
+        // Edge detection
+        Imgproc.Canny(original, edgesDetected, 50, 200, 3, false);
+        //Copy edges to the images that will display the results in BGR
+        Imgproc.cvtColor(edgesDetected, edgesDetectedRGB, Imgproc.COLOR_GRAY2BGR);
+        // Probabilistic Line Transform
+        Mat linesP = new Mat(); // will hold the results of the detection
+        double minLineLength = edgesDetectedRGB.size().width/4;
+        Imgproc.HoughLinesP(edgesDetected, linesP, 1, Math.PI / 180, 50, minLineLength, 10);// runs the actual detection
+        // Draw the lines
+        
+        for (int x = 0; x < linesP.rows(); x++) {
+            double[] l = linesP.get(x, 0);
+            //New angles
+            Point p1 = new Point(l[0], l[1]);
+            Point p2 = new Point(l[2], l[3]);
+            double m = Math.abs(p2.y - p1.y)/(p2.x - p1.x);
+            //System.out.println(l[0]);
+            //System.out.println(l[1]);
+            //System.out.println(l[2]);
+            //System.out.println(l[3]);
+            if(m<0.1) {
+                //System.out.println("m: " + m);
+                Imgproc.line(edgesDetectedRGB, new Point(l[0], l[1]), new Point(l[2], l[3]), new Scalar(0, 0, 255), 3, Imgproc.LINE_AA, 0);
+            }
+        }
+        
+        //Convert MAT into a BufferedImage
+        BufferedImage toBeClassifiedImg = toBufferedImage(edgesDetectedRGB);
+        //Calculate if its sheet music or not
+        isSheetMusic = classifier_HoughLinesP(toBeClassifiedImg);  
+            
+    }
+    catch(Exception e){
+            System.err.println(e);
+        }
+        return isSheetMusic;
+ }
+ 
  //******************
  //INTERNAL FUNCTIONS
  //******************
-    private static boolean classifier_HoughLineP(BufferedImage img){
+    private static boolean classifier_HoughLinesP(BufferedImage img){
     System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
     try {
@@ -143 +183 @@
         percentage = ((float)redCount/(float)pixelCount)*(float)100;
         //If more than %10 and less than %50 then its sheet music!
-        if(percentage > 10 && percentage < 50){
+        if(percentage > CLASSIFIER_HOUGHLINESP_MIN && percentage < CLASSIFIER_HOUGHLINESP_MAX){
             return true;}
         }