Changeset 640

Timestamp:

01/31/10 22:02:02 (2 years ago)

Author:

mosttw

Message:

Modified sobel.py to use HSV space

Files:

branches/2010-image-rec/sobel.py

r632	r640
5	5
6	6	fn = sys.argv[1]
	7	try:
	8	use_hsv = sys.argv[2] == 'hsv'
	9	except IndexError:
	10	use_hsv = False
	11
7	12	im = cv.LoadImage(fn)
	13	if use_hsv:
	14	hsv = cv.CreateImage((im.width, im.height), cv.IPL_DEPTH_8U, 3)
	15	cv.CvtColor(im, hsv, cv.CV_BGR2HSV)
	16	im = hsv
8	17
9	18	print "sobelizing"
…	…
14	23	sobel_img = cv.CreateImage((im.width, im.height), cv.IPL_DEPTH_8U, 3)
15	24	cv.ConvertScale(sobel, sobel_img, 8.0)
16		cv.SaveImage(sys.argv[1] + '-sobel.png', sobel_img)
	25	if use_hsv:
	26	sobel_fn = fn + '-hsv-sobel.png'
	27	else:
	28	sobel_fn = fn + '-sobel.png'
	29
	30	cv.SaveImage(sobel_fn, sobel_img)
17	31	sys.exit(0)
18	32
…	…
20	34	#median = cv.CreateImage((im.width, im.height), cv.IPL_DEPTH_8U, 3)
21	35	#cv.Smooth(sobel_img, median, cv.CV_MEDIAN, 7)
22		#cv.SaveImage(~~sys.argv[1]~~ + '-median.png', median)
	36	#cv.SaveImage(fn + '-median.png', median)
23	37