working feature extraction + knearest neighbour

2013-10-15 01:29:31 +02:00 · 2013-10-15 01:29:31 +02:00 · 266ce7c5b0
commit 266ce7c5b0
parent 4b2bb43c62
139 changed files with 158 additions and 30 deletions
--- a/.gitignore
+++ b/.gitignore
@ -0,0 +1 @@
+*haar*
--- a/cat.png
+++ b/cat.png
--- a/cat2.png
+++ b/cat2.png
--- a/dir.py
+++ b/dir.py
@ -0,0 +1,9 @@
+
+import os
+
+cat_dir = "img/cat"
+dog_dir = "img/dog"
+
+for directory in [cat_dir, dog_dir]:
+	for fn in os.listdir(directory):
+		print directory, fn
--- a/dog.png
+++ b/dog.png
--- a/dog2.png
+++ b/dog2.png
--- a/facer.rb
+++ b/facer.rb
@ -0,0 +1,41 @@
+require 'rubygems'
+require 'opencv'
+
+face_detectors = %w(
+  /usr/local/share/opencv/haarcascades/haarcascade_frontalface_default.xml
+).collect { |fd| OpenCV::CvHaarClassifierCascade::load(fd) }
+
+
+file_names =  Dir.glob(ARGV[0] || "/Users/sandro/Ruby/image_labs/lib/facer/euruko2/*.jpg") 
+
+file_names.each_with_index do |file_name,i|
+  
+  face_rectangles = []
+  opencv_load = OpenCV::IplImage.load(file_name) 
+  face_detectors.each_with_index do |face_detector,k|
+    face_detector.detect_objects(opencv_load).each do |rect|
+      face_rectangles << [ 
+        rect.top_left.x,      rect.top_left.y,  
+        rect.top_right.x    - rect.top_left.x,  
+        rect.bottom_left.y  - rect.top_left.y 
+      ]
+    end
+  end
+
+  File.open("#{File.basename(file_name)}.html","w") do |file|
+    file.write <<-HTML
+      <html>
+        <body style="margin:0px; background: url('file://#{file_name}') no-repeat top left;">
+          #{  face_rectangles.collect do |(x,y,w,h)|
+                "<div class=\"face\" style=\"position: absolute; left: #{x}px; 
+                  top: #{y}px; width: #{w}px; height: #{h}px; border: 2px solid red; \"></div>"
+              end.join("\n") }
+        </body>
+      </html>
+    HTML
+  end
+  
+  puts "End #{file_name}"
+
+end
+
--- a/find_lines.py
+++ b/find_lines.py
@ -1,42 +1,70 @@
 #!/usr/bin/env python

-import numpy as np
-from math import pi
-import cv2
-import cv
-import sys
+import cv2, cv, sys, math, os, numpy
+from scipy.spatial import KDTree

-if len(sys.argv) > 1:
-	fn = sys.argv[1]
-	print 'loading %s ...' % fn
-	img1 = cv2.imread(fn, 0)
-	img = cv.LoadImage(fn, cv.CV_LOAD_IMAGE_GRAYSCALE)
-	size = cv.GetSize(img)
+def extractFeatures(label):

-	temp = cv.CreateImage(size, img.depth, img.nChannels)
-	print temp
-	cv.Smooth(img, temp)
+	directory = "img/" + label + "/"

-	canny = cv2.Canny(temp, 50, 100)
-	color_dst = cv2.cvtColor(canny, cv2.COLOR_GRAY2BGR)
-	lines = cv2.HoughLinesP(canny, 1, pi/90, 20, np.array([]), 5)
+	features = []

-	try:
-		for line in lines[0]:
-			cv2.line(color_dst, (line[0], line[1]), (line[2], line[3]), cv.RGB(255,0,0), 1, 8)
-	except:
-		pass
+	for fn in os.listdir(directory):

-	print lines[0].size
+		img = cv2.imread(directory + fn, 0)

-	cv2.namedWindow("Original")
-	cv2.imshow("Original", img)
+		#temp = cv.CreateImage((100,100), cv.CV_8U, 1)
+		#cv.Smooth(img, temp)

-	cv2.namedWindow('Lines image')
-	cv2.imshow('Lines image', color_dst)
+		canny = cv2.Canny(img, 50, 100)
+		color_dst = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)

-	cv2.waitKey()
+		# find colored
+		black_pixels = numpy.count_nonzero(img)
+
+		# find lines lines
+		lines = cv2.HoughLinesP(canny, 1, math.pi/360, 5, None, 10, 1)
+
+		lengths = []
+		angles = []
+		try:
+			for line in lines[0]:
+				x1, y1, x2, y2 = line
+				#cv2.line(color_dst, (x1, y1), (x2, y2), cv.RGB(255,0,0), 1, 8)
+				length = int(math.sqrt(math.pow((x1-x2), 2) + math.pow((y1-y2), 2)))
+				lengths.append(length)
+
+				angle = int(math.degrees(math.atan((y1-y2) / (x1-x2))))
+				angles.append(angle)
+		except:
+			pass
+
+		# print out everything
+		lines_count = len(lengths)
+		mid_length = sum(lengths) / lines_count
+		mid_angle = sum(angles) / lines_count
+
+		features.append([[lines_count, mid_length, mid_angle, black_pixels], label])
+
+		#cv2.namedWindow("Original")
+		#cv2.imshow("Original", img)
+
+		#cv2.namedWindow('Lines image ' + fn)
+		#cv2.imshow('Lines image ' + fn, color_dst)
+
+	return features


-else:
-	print "Please give a image path"
+if __name__ == "__main__":
+	arr = extractFeatures("cat") + extractFeatures("dog")
+	test_label = arr[0][1]
+	test_feature = arr[0][0]
+	labels = map(lambda a: a[1], arr)[1:]
+	features = map(lambda a: a[0], arr)[1:]
+
+	tree = KDTree(features)
+	d, i = tree.query(test_feature)
+
+
+	print test_label + " is predicted to be a " + labels[i]
+
--- a/img/.DS_Store
+++ b/img/.DS_Store
--- a/img/cat/1.png
+++ b/img/cat/1.png
--- a/img/cat/10.png
+++ b/img/cat/10.png
--- a/img/cat/11.png
+++ b/img/cat/11.png
--- a/img/cat/12.png
+++ b/img/cat/12.png
--- a/img/cat/13.png
+++ b/img/cat/13.png
--- a/img/cat/14.png
+++ b/img/cat/14.png
--- a/img/cat/15.png
+++ b/img/cat/15.png
--- a/img/cat/16.png
+++ b/img/cat/16.png
--- a/img/cat/17.png
+++ b/img/cat/17.png
--- a/img/cat/18.png
+++ b/img/cat/18.png
--- a/img/cat/19.png
+++ b/img/cat/19.png
--- a/img/cat/2.png
+++ b/img/cat/2.png
--- a/img/cat/20.png
+++ b/img/cat/20.png
--- a/img/cat/21.png
+++ b/img/cat/21.png
--- a/img/cat/22.png
+++ b/img/cat/22.png
--- a/img/cat/23.png
+++ b/img/cat/23.png
--- a/img/cat/24.png
+++ b/img/cat/24.png
--- a/img/cat/25.png
+++ b/img/cat/25.png
--- a/img/cat/26.png
+++ b/img/cat/26.png
--- a/img/cat/27.png
+++ b/img/cat/27.png
--- a/img/cat/28.png
+++ b/img/cat/28.png
--- a/img/cat/29.png
+++ b/img/cat/29.png
--- a/img/cat/3.png
+++ b/img/cat/3.png
--- a/img/cat/30.png
+++ b/img/cat/30.png
--- a/img/cat/31.png
+++ b/img/cat/31.png
--- a/img/cat/32.png
+++ b/img/cat/32.png
--- a/img/cat/33.png
+++ b/img/cat/33.png
--- a/img/cat/34.png
+++ b/img/cat/34.png
--- a/img/cat/35.png
+++ b/img/cat/35.png
--- a/img/cat/36.png
+++ b/img/cat/36.png
--- a/img/cat/37.png
+++ b/img/cat/37.png
--- a/img/cat/38.png
+++ b/img/cat/38.png
--- a/img/cat/39.png
+++ b/img/cat/39.png
--- a/img/cat/4.png
+++ b/img/cat/4.png
--- a/img/cat/40.png
+++ b/img/cat/40.png
--- a/img/cat/41.png
+++ b/img/cat/41.png
--- a/img/cat/42.png
+++ b/img/cat/42.png
--- a/img/cat/43.png
+++ b/img/cat/43.png
--- a/img/cat/44.png
+++ b/img/cat/44.png
--- a/img/cat/45.png
+++ b/img/cat/45.png
--- a/img/cat/46.png
+++ b/img/cat/46.png
--- a/img/cat/47.png
+++ b/img/cat/47.png
--- a/img/cat/48.png
+++ b/img/cat/48.png
--- a/img/cat/49.png
+++ b/img/cat/49.png
--- a/img/cat/5.png
+++ b/img/cat/5.png
--- a/img/cat/50.png
+++ b/img/cat/50.png
--- a/img/cat/51.png
+++ b/img/cat/51.png
--- a/img/cat/52.png
+++ b/img/cat/52.png
--- a/img/cat/54.png
+++ b/img/cat/54.png
--- a/img/cat/55.png
+++ b/img/cat/55.png
--- a/img/cat/56.png
+++ b/img/cat/56.png
--- a/img/cat/57.png
+++ b/img/cat/57.png
--- a/img/cat/58.png
+++ b/img/cat/58.png
--- a/img/cat/59.png
+++ b/img/cat/59.png
--- a/img/cat/6.png
+++ b/img/cat/6.png
--- a/img/cat/60.png
+++ b/img/cat/60.png
--- a/img/cat/61.png
+++ b/img/cat/61.png
--- a/img/cat/62.png
+++ b/img/cat/62.png
--- a/img/cat/63.png
+++ b/img/cat/63.png
--- a/img/cat/64.png
+++ b/img/cat/64.png
--- a/img/cat/65.png
+++ b/img/cat/65.png
--- a/img/cat/7.png
+++ b/img/cat/7.png
--- a/img/cat/8.png
+++ b/img/cat/8.png
--- a/img/cat/9.png
+++ b/img/cat/9.png
--- a/img/dog/1.png
+++ b/img/dog/1.png
--- a/img/dog/10.png
+++ b/img/dog/10.png
--- a/img/dog/11.png
+++ b/img/dog/11.png
--- a/img/dog/12.png
+++ b/img/dog/12.png
--- a/img/dog/13.png
+++ b/img/dog/13.png
--- a/img/dog/14.png
+++ b/img/dog/14.png
--- a/img/dog/15.png
+++ b/img/dog/15.png
--- a/img/dog/16.png
+++ b/img/dog/16.png
--- a/img/dog/17.png
+++ b/img/dog/17.png
--- a/img/dog/18.png
+++ b/img/dog/18.png
--- a/img/dog/19.png
+++ b/img/dog/19.png
--- a/img/dog/2.png
+++ b/img/dog/2.png
--- a/img/dog/20.png
+++ b/img/dog/20.png
--- a/img/dog/21.png
+++ b/img/dog/21.png
--- a/img/dog/22.png
+++ b/img/dog/22.png
--- a/img/dog/23.png
+++ b/img/dog/23.png
--- a/img/dog/24.png
+++ b/img/dog/24.png
--- a/img/dog/25.png
+++ b/img/dog/25.png
--- a/img/dog/26.png
+++ b/img/dog/26.png
--- a/img/dog/27.png
+++ b/img/dog/27.png
--- a/img/dog/28.png
+++ b/img/dog/28.png
--- a/img/dog/29.png
+++ b/img/dog/29.png
--- a/img/dog/3.png
+++ b/img/dog/3.png
--- a/img/dog/30.png
+++ b/img/dog/30.png
--- a/img/dog/31.png
+++ b/img/dog/31.png
--- a/img/dog/32.png
+++ b/img/dog/32.png
--- a/img/dog/33.png
+++ b/img/dog/33.png
--- a/Show more
+++ b/Show more