def pretreat_img(fileName):
if endwith(fileName, '.jpg', ',JPG', '.PNG', '.png'):
img = cv.imread(fileName)
img, flag = getFaceImage(img)
if flag == 1:
img = warpAffineImg(img)
return img
else:
return False
def pretreat_img_chinese(fileName):
if endwith(fileName, '.jpg', ',JPG', '.PNG', '.png'):
img = cv.imdecode(np.fromfile(fileName, dtype=np.uint8), -1)
img, flag = getFaceImage(img)
if flag == 1:
img = warpAffineImg(img)
return img
else:
return False
def pretreat_img(fileName):
if endwith(fileName, '.jpg', ',JPG', '.PNG', '.png'):
img = cv.imread(fileName)
if img is None: # 图片为空
return 0, 0
else: # 图片不为空，进行预处理
# 图片文件过大，计算机内存不足，滤过过大文件
if img.shape[0] < 1500 and img.shape[1] < 1500:
img, flag = getFaceImage(img) # 人脸识别
if flag == 1:
img = warpAffineImg(img) # 人脸对齐
return img, flag # 成功获取目标图片
# 未能识别人脸
else:
return img, 0
# 文件名非指定图片格式
else:
return 0, 0
def pretreat_img_chinese(fileName):
# 后缀名判断
if endwith(fileName, '.jpg', ',JPG', '.PNG', '.png'):
# 对路径包含中文字符的文件进行读取
img = cv.imdecode(np.fromfile(fileName, dtype=np.uint8), -1)
if img is None: # 图片为空
return 0, 0
else: # 图片不为空，进行预处理
# 图片文件过大，计算机内存不足，滤过过大文件
if img.shape[0] < 1500 and img.shape[1] < 1500:
img, flag = getFaceImage(img) # 人脸识别
if flag == 1:
img = warpAffineImg(img) # 人脸对齐
return img, flag # 成功获取目标图片
# 未能识别人脸
else:
return img, 0
# 文件名非指定图片格式
else:
return 0, 0
def pretreat_img_internet(fileName):
# 后缀名判断
if endwith(fileName, '.jpg', ',JPG', '.PNG', '.png'):
# 从网络路径读取图片文件
cap = cv.VideoCapture(fileName)
ret, img = cap.read()
if img is None: # 图片为空
return 0, 0
else: # 图片不为空，进行预处理
# 图片文件过大，计算机内存不足，滤过过大文件
if img.shape[0] < 1500 and img.shape[1] < 1500:
img, flag = getFaceImage(img) # 人脸识别
if flag == 1:
img = warpAffineImg(img) # 人脸对齐
return img, flag # 成功获取目标图片
# 未能识别人脸
else:
return img, 0
# 文件名非指定图片格式
else:
return 0, 0