百度图片增强

salai/common_config.py

@@ -18,10 +18,13 @@ SAVE_PATH = config.get('oss', 'SAVE_PATH')
 # 账号配置
 TOKEN = config.get('user', 'TOKEN')
 
-
 # redis配置
 REDIS_HOST = config.get('redis', 'HOST')
 REDIS_PORT = config.get('redis', 'PORT')
 REDIS_PASSWORD = config.get('redis', 'PASSWORD')
 REDIS_DB = config.get('redis', 'DB')
 
+# 百度配置
+BAI_DU_APP_ID = config.get('baidu', 'APP_ID')
+BAI_DU_API_KEY = config.get('baidu', 'API_KEY')
+BAIDU_API_SECRET_KEY = config.get('baidu', 'SECRET_KEY')

salai/env.example.ini

@@ -16,3 +16,8 @@ HOST=
 PORT=
 PASSWORD=
 DB=
+
+[baidu]
+APP_ID=
+API_KEY=
+SECRET_KEY=
\ No newline at end of file

salai/helper.py

 from PIL import Image
 import requests
 import os
-from salai.common_config import DOWNLOAD_IMAGE_PATH, ACCESS_KEY_ID, ACCESS_KEY_SECRET, ENDPOINT, BUCKET_NAME, SAVE_PATH
+from salai.common_config import DOWNLOAD_IMAGE_PATH, ACCESS_KEY_ID, ACCESS_KEY_SECRET, ENDPOINT, BUCKET_NAME, SAVE_PATH, BAIDU_API_SECRET_KEY, BAI_DU_API_KEY, BAI_DU_APP_ID
 import hashlib
 from oss2 import Auth, Bucket
-
+from aip import AipImageProcess
+import base64
+from loguru import logger
 
 def split_image(image_file):
     """
@@ -55,7 +57,7 @@ async def download_image(url: str, filename: str, index: int = 0):
     save_image_list = list()
 
     filename = get_hash(filename) + ".png"
-    path = f"{DOWNLOAD_IMAGE_PATH}{filename}"
+    path = f"{DOWNLOAD_IMAGE_PATH}/{filename}"
     with open(path, 'wb') as f:
         f.write(r.content)
         f.close()
@@ -87,6 +89,26 @@ async def download_image(url: str, filename: str, index: int = 0):
     result = list()
     if len(save_image_list) > 0:
         for file in save_image_list:
+            if index > 0:
+                client = AipImageProcess(BAI_DU_APP_ID, BAI_DU_API_KEY, BAIDU_API_SECRET_KEY)
+
+                with open(file['path'], 'rb') as fp:
+                     image = fp.read()
+
+                # 图像增强
+                logger.info("图像增强")
+                data = client.imageDefinitionEnhance(image)
+                imgdata = base64.b64decode(data["image"])
+                with open(file['path'], 'wb') as f:
+                    f.write(imgdata)
+
+                # 无损放大
+                logger.info("无损放大")
+                data = client.imageQualityEnhance(image)
+                imgdata = base64.b64decode(data["image"])
+                with open(file['path'], 'wb') as f:
+                    f.write(imgdata)
+
             save_name = f"{SAVE_PATH}/{file['name']}"
             await oss_upload(save_name, file['path'])
             save_name = "https://{}.{}/{}".format(BUCKET_NAME, ENDPOINT, save_name)