aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat

Browse Source 
fix   relight 补光接口修改
This commit is contained in:
zhouchengrong
2024-09-03 15:49:42 +08:00
parent d8883f6eb6
commit f5fc6b0c68
2 changed files with 3 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app/api/api_brighten.py
View File

@@ -26,15 +26,11 @@ async def brighten(request_item: BrightenModel):
创建一个具有以下参数的请求体: 创建一个具有以下参数的请求体:
- **image_url**: 提亮图片url - **image_url**: 提亮图片url
- **brighten_value**: 提高亮度的比重 亮度因子 1.0 表示原始亮度，1.5 表示增加 50% 的亮度 - **brighten_value**: 提高亮度的比重 亮度因子 1.0 表示原始亮度，1.5 表示增加 50% 的亮度
- **brighten_bucket**: 结果图保存的bucket
- **brighten_name**: 结果图保存的object name
示例参数： 示例参数：
{ {
"image_url": "aida-users/89/relight_image/3850e17b-3efd-4597-90ef-2a7bcd1a1a0b-0-89.png", "image_url": "aida-users/89/relight_image/3850e17b-3efd-4597-90ef-2a7bcd1a1a0b-0-89.png",
"brighten_value": 1.5, "brighten_value": 1.5
"brighten_bucket": "test",
"brighten_name": "12341556-89.jpg"
} }
""" """
try: try:
@@ -46,7 +42,7 @@ async def brighten(request_item: BrightenModel):
new_image.save(image_data, format='PNG') new_image.save(image_data, format='PNG')
image_data.seek(0) image_data.seek(0)
image_bytes = image_data.read() image_bytes = image_data.read()
req = oss_upload_image(bucket=request_item.brighten_bucket, object_name=request_item.brighten_name, image_bytes=image_bytes) req = oss_upload_image(bucket=request_item.image_url.split('/')[0], object_name=request_item.image_url[request_item.image_url.find('/') + 1:], image_bytes=image_bytes)
brighten_url = f"{req.bucket_name}/{req.object_name}" brighten_url = f"{req.bucket_name}/{req.object_name}"
logger.info(f"run time is : {time.time() - start_time}") logger.info(f"run time is : {time.time() - start_time}")
except Exception as e: except Exception as e:
@@ -57,9 +53,7 @@ async def brighten(request_item: BrightenModel):
if __name__ == '__main__': if __name__ == '__main__':
request_item = BrightenModel(image_url="aida-users/89/relight_image/3850e17b-3efd-4597-90ef-2a7bcd1a1a0b-0-89.png", request_item = BrightenModel(image_url="aida-users/89/relight_image/3850e17b-3efd-4597-90ef-2a7bcd1a1a0b-0-89.png",
brighten_value=1.5, brighten_value=1.5)
brighten_bucket="test",
brighten_name="12341556-89.png")
image = oss_get_image(bucket=request_item.image_url.split('/')[0], object_name=request_item.image_url[request_item.image_url.find('/') + 1:], data_type="PIL") image = oss_get_image(bucket=request_item.image_url.split('/')[0], object_name=request_item.image_url[request_item.image_url.find('/') + 1:], data_type="PIL")
new_image = increase_brightness(image, request_item.brighten_value) new_image = increase_brightness(image, request_item.brighten_value)
new_image.show() new_image.show()

2
app/schemas/brighten.py
View File

@@ -4,5 +4,3 @@ from pydantic import BaseModel
class BrightenModel(BaseModel): class BrightenModel(BaseModel):
image_url: str image_url: str
brighten_value: float brighten_value: float
brighten_bucket: str
brighten_name: str