import cv2
import sys
import os.path
def detect(filename, cascade_file = "./lbpcascade_animeface.xml"):
if not os.path.isfile(cascade_file):
raise RuntimeError("%s: not found" % cascade_file)
cascade = cv2.CascadeClassifier(cascade_file)
image = cv2.imread(filename, cv2.IMREAD_COLOR)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
gray = cv2.equalizeHist(gray)
faces = cascade.detectMultiScale(gray,
# detector options
scaleFactor = 1.1,
minNeighbors = 5,
minSize = (24, 24))
for (x, y, w, h) in faces:
cv2.rectangle(image, (x, y), (x + w, y + h), (0, 0, 255), 2)
for i,(x,y,w,h) in enumerate(faces):
face=image[y:y+h,x:x+w,:]
face=cv2.resize(face,(96,96))
save_filename='%s-%d.jpg'%(os.path.basename(filename).split('.')[0],i)
cv2.imwrite('faces/'+save_filename,face)
cv2.imshow("AnimeFaceDetect", image)
cv2.waitKey(0)
cv2.imwrite("out.png", image)
if len(sys.argv) != 2:
sys.stderr.write("usage: detect.py <filename>\n")
sys.exit(-1)
detect(sys.argv[1])
sys.argv,其实就是一个list,它是sys模块下的一个全局变量,第一个元素是模块名、后面是依次传入的参数,比如可以这样传入 pyton temp.py a b c d,一共传入a、b、c、d四个参数
len(sys.argv) == 5
那么sys.argv[0] == “temp.py” sys.argv[1] == “a” sys.argv[2] == “b”
参考:
GAN学习指南:从原理入门到制作生成Demo
https://blog.csdn.net/cadi2011/article/details/84638659