Created
August 3, 2016 15:33
-
-
Save jhh/3a6c56a7ca7dbe4f90ee80c252eb7537 to your computer and use it in GitHub Desktop.
Solution to exercise 2.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import cv2 | |
import sys | |
import os | |
def simple_threshold(path, thresh): | |
# read in the image | |
img = cv2.imread(path) | |
# display the image and wait for key press | |
cv2.imshow('original', img) | |
cv2.waitKey(0) | |
# http://docs.opencv.org/2.4/modules/imgproc/doc/miscellaneous_transformations.html#cvtcolor | |
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) | |
# http://docs.opencv.org/2.4/modules/imgproc/doc/filtering.html#gaussianblur | |
blurred = cv2.GaussianBlur(gray, (5, 5), 0) | |
# http://docs.opencv.org/2.4/modules/imgproc/doc/miscellaneous_transformations.html#threshold | |
(T, thresholded) = cv2.threshold(blurred, thresh, 255, cv2.THRESH_BINARY) | |
# display the image and wait for key press | |
cv2.imshow('thresholded image', thresholded) | |
cv2.waitKey(0) | |
# usage: python ex_002.py path/to/image.jpg threshold | |
# example: python ex_002.py ~/Projects/sf/notebooks/RealFullField/245.jpg 99 | |
if __name__ == '__main__': | |
if len(sys.argv) < 3: | |
print("usage: python ex_002.py path/to/image.jpg threshold") | |
sys.exit() | |
imgPath = sys.argv[1] | |
if not os.path.isfile(imgPath): | |
print("Image file not found.") | |
sys.exit() | |
threshold = sys.argv[2] | |
if not threshold.isdigit(): | |
print("Enter a digit for threshold") | |
sys.exit() | |
simple_threshold(imgPath, float(threshold)) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment