parent
9dd9260a21
commit
654fb4952c
@ -0,0 +1,7 @@ |
|||||||
|
certifi==2021.5.30 |
||||||
|
charset-normalizer==2.0.4 |
||||||
|
colorama==0.4.4 |
||||||
|
idna==3.2 |
||||||
|
requests==2.26.0 |
||||||
|
tqdm==4.62.2 |
||||||
|
urllib3==1.26.6 |
@ -0,0 +1,88 @@ |
|||||||
|
desc = ''' |
||||||
|
*********************************************** |
||||||
|
authOr: newrain * |
||||||
|
blog: https://blog.csdn.net/NewRain_wang * |
||||||
|
https://newrain001.gitee.io * |
||||||
|
github: https://github.com/newrain001 * |
||||||
|
gitee : https://gitee.com/newrain001 * |
||||||
|
email : newrain_wang@163.com * |
||||||
|
*********************************************** |
||||||
|
1、此脚本采用旷视ai人脸美颜技术接口 |
||||||
|
https://www.faceplusplus.com.cn/ |
||||||
|
2、文件不得大于2M |
||||||
|
3、像素不大大于4096x4096 |
||||||
|
4、首次执行程序后会生成配置文件,请注册并填写api_key和api_secret |
||||||
|
5、配置文件解析 |
||||||
|
'whitening': 100, # 美白 |
||||||
|
'smoothing': 100, # 磨皮 |
||||||
|
'thinface': 100, # 瘦脸 |
||||||
|
'shrink_face': 100, # 小脸 |
||||||
|
'enlarge_eye': 100, # 大眼 |
||||||
|
'remove_eyebrow': 100, # 去眉毛 |
||||||
|
'filter_type': 'beautify', # 滤镜 https://console.faceplusplus.com.cn/documents/134252584 |
||||||
|
6、原图请放置在images文件夹中,新图会在源文件前加上new |
||||||
|
''' |
||||||
|
from requests import post |
||||||
|
from time import sleep |
||||||
|
from json import dump,load |
||||||
|
import os |
||||||
|
import base64 |
||||||
|
|
||||||
|
|
||||||
|
# from tqdm import tqdm |
||||||
|
|
||||||
|
class Kuangshi(): |
||||||
|
def __init__(self): |
||||||
|
if not os.path.exists('./readme.txt'): |
||||||
|
with open('./readme.txt', 'w', encoding='utf-8') as f: |
||||||
|
f.write(desc) |
||||||
|
os.system('notepad ./readme.txt') |
||||||
|
if not os.path.exists('./config.json'): |
||||||
|
config = { |
||||||
|
'api_key': '', |
||||||
|
'api_secret': '', |
||||||
|
'whitening': 100, # 美白 |
||||||
|
'smoothing': 100, # 磨皮 |
||||||
|
'thinface': 100, # 瘦脸 |
||||||
|
'shrink_face': 100, # 小脸 |
||||||
|
'enlarge_eye': 100, # 大眼 |
||||||
|
'remove_eyebrow': 100, # 去眉毛 |
||||||
|
'filter_type': 'beautify', # 滤镜 https://console.faceplusplus.com.cn/documents/134252584 |
||||||
|
} |
||||||
|
with open('./config.json', 'w') as f: |
||||||
|
dump(config, f, indent='\t', ensure_ascii=False) |
||||||
|
print('请先修改配置文件运行') |
||||||
|
sleep(5) |
||||||
|
else: |
||||||
|
with open('./config.json', 'r') as f: |
||||||
|
self.config = load(f) |
||||||
|
if not os.path.exists('./images'): |
||||||
|
os.mkdir('./images') |
||||||
|
print('请将需要优化的图片放置在images目录中') |
||||||
|
sleep(5) |
||||||
|
else: |
||||||
|
image = os.listdir('./images') |
||||||
|
self.image = [i for i in image if not i.startswith('new')] |
||||||
|
|
||||||
|
def beautify_image(self): |
||||||
|
os.startfile('images') |
||||||
|
for image in self.image: |
||||||
|
#try: |
||||||
|
if not image.startswith('new'): |
||||||
|
r = post(url='https://api-cn.faceplusplus.com/facepp/v2/beautify', |
||||||
|
data=self.config, |
||||||
|
files={ |
||||||
|
'image_file': (image, open(os.path.join('./images', image), 'rb'), 'image/jpeg')}) |
||||||
|
print(r.json()) |
||||||
|
with open(os.path.join('./images', 'new_' + image), 'wb') as f: |
||||||
|
f.write(base64.b64decode(r.json()['result'])) |
||||||
|
sleep(3) |
||||||
|
# except Exception as e: |
||||||
|
# print(e) |
||||||
|
# print('处理异常') |
||||||
|
# sleep(5) |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
ai = Kuangshi() |
||||||
|
ai.beautify_image() |
Loading…
Reference in new issue