29 lines
831 B
29 lines
831 B
# USAGE - How to run this code ? |
|
# python find_shapes.py --image shapes.png |
|
#python findredlinedpolygonfromgooglemaps.py --image stanford.png |
|
|
|
import numpy as np |
|
import argparse |
|
import cv2 |
|
|
|
# construct the argument parse and parse the arguments |
|
ap = argparse.ArgumentParser() |
|
ap.add_argument("-i", "--image", help = "path to the image file") |
|
args = vars(ap.parse_args()) |
|
|
|
# load the image |
|
image = cv2.imread(args["image"]) |
|
lower = np.array([20,0,155]) |
|
upper = np.array([255,120,250]) |
|
shapeMask = cv2.inRange(image, lower, upper) |
|
|
|
# find the contours in the mask |
|
(cnts, _) = cv2.findContours(shapeMask.copy(), cv2.RETR_EXTERNAL, |
|
cv2.CHAIN_APPROX_SIMPLE) |
|
cv2.imshow("Mask", shapeMask) |
|
|
|
# loop over the contours |
|
for c in cnts: |
|
cv2.drawContours(image, [c], -1, (0, 255, 0), 2) |
|
cv2.imshow("Image", image) |
|
cv2.waitKey(0) |