new

Author: StevenHuang2020

README.md

@@ -11,3 +11,7 @@ Facial key-points detection by using CNN model. Dataset：FG-Net
 <img src="images/Figure_4.png" width="320" height="240">
 <img src="images/Figure_5.png" width="320" height="240">
 <img src="images/Figure_2.png" width="320" height="240">
+<img src="images/Figure_7.png" width="320" height="240">
+<img src="images/Figure_11.png" width="320" height="240">
+<img src="images/Figure_12.png" width="320" height="240">
+<img src="images/Figure_13.png" width="320" height="240">

commonModule/ImageBase.py

@@ -5,8 +5,9 @@
 from matplotlib import pyplot as plt
 
 def changeBgr2Rbg(img): #input color img
- b,g,r = cv2.split(img) # get b,g,r
- img = cv2.merge([r,g,b])
+ if getImagChannel(img) == 3:
+ b,g,r = cv2.split(img) # get b,g,r
+ img = cv2.merge([r,g,b])
  return img
 
 def loadImg(file,mode=cv2.IMREAD_COLOR):

faceIdentification.py

@@ -21,7 +21,7 @@ def CascadeDetect(cascPath=r'./res/haarcascade_frontalface_default.xml'):
 
 def main():
  cv2.useOptimized()
- file = r'./res/obama.jpg'#r'./res/Lenna.png' #
+ file = r'./res/trump.jpg'#r'./res/Lenna.png' #obama.jpg
 
  print('Number of parameter:', len(sys.argv))
  print('Parameters:', str(sys.argv))
@@ -32,11 +32,15 @@ def main():
  img = loadImg(file,mode=cv2.IMREAD_COLOR) # IMREAD_GRAYSCALE IMREAD_COLOR
 
  faceROI = CascadeDetect()
- 
  faceR=faceROI.getDetectImg(img)
  face = faceROI.detecvFaceImgOne(img)
  faceGray = cv2.cvtColor(face, cv2.COLOR_BGR2GRAY)
 
+ img = changeBgr2Rbg(img)
+ faceR = changeBgr2Rbg(faceR)
+ #face = changeBgr2Rbg(face)
+ #faceGray = changeBgr2Rbg(faceGray)
+ 
  ls,names = [],[]
  ls.append(img),names.append('orignal')
  ls.append(faceR),names.append('faceR')
@@ -46,7 +50,7 @@ def main():
  plotImagList(ls,names)
 
  #faceGray = resizeImg(faceGray,newW,newH)
- #img.writeImg(face,'./res/myface_.png')
+ writeImg(face,'./res/myface_.png')
  #writeImg(faceGray,'./res/myface_gray.png')
 
 if __name__=="__main__":

images/Capture.PNG

@@ -19,7 +19,11 @@ def argCmdParse():
 def preditImg(img, modelName = r'./weights/trainFacialRecognition.h5'): # #r'./weightLoc/trainFacialRecognition.h5'
  model = ks.models.load_model(modelName)
  print(img.shape)
- x = img[:,:,0]
+ if getImagChannel(img) == 3:
+ x = img[:,:,0]
+ else:
+ x = img
+ 
  print(x.shape)
  x = x.reshape((1,x.shape[0],x.shape[1],1))
  print(x.shape)

images/Figure_11.png

@@ -36,7 +36,7 @@ def predictPerson(feature):
 
 def main():
  arg = argCmdParse() 
- file = r'./res/001A29_ex2.jpg' #r'./res/001A29.jpg' #arg.source #
+ file = r'./res/myface_.png' #r'./res/001A29_ex2.jpg' #r'./res/001A29.jpg' #arg.source #
 
  img = loadImg(file)
  img = resizeImg(img,newW,newH)