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

feat relight 增加打光方向

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-07-03 10:26:40 +08:00
parent 72428a73ab
commit 48eaa9c2e3
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/api_generate_image.py
View File

@@ -151,12 +151,14 @@ def generate_relight_image(request_item: GenerateRelightImageModel, background_t
- **tasks_id**: 任务id 用于取消生成任务和获取生成结果 - **tasks_id**: 任务id 用于取消生成任务和获取生成结果
- **prompt**: 想要生成图片的描述词 - **prompt**: 想要生成图片的描述词
- **image_url**: 被生成图片的S3或minio url地址 - **image_url**: 被生成图片的S3或minio url地址
- **direction**: 光源方向 Right Light Left Light Top Light Bottom Light
示例参数： 示例参数：
{ {
"tasks_id": "123-89", "tasks_id": "123-89",
"prompt": "beautiful woman, detailed face, sunshine, outdoor, warm atmosphere", "prompt": "beautiful woman, detailed face, sunshine, outdoor, warm atmosphere",
"image_url": "aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png" "image_url": "aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png",
"direction": "Right Light"
} }
""" """
try: try:

1
app/schemas/generate_image.py
View File

@@ -27,3 +27,4 @@ class GenerateRelightImageModel(BaseModel):
tasks_id: str tasks_id: str
prompt: str prompt: str
image_url: str image_url: str
direction: str

9
app/service/generate_image/service_generate_relight_image.py
View File

@@ -7,16 +7,15 @@
@Date ：2023/7/26 12:01:05 @Date ：2023/7/26 12:01:05
@detail ： @detail ：
""" """
import io
import json import json
import logging import logging
import time import time
import cv2 import cv2
import numpy as np
import redis import redis
import tritonclient.grpc as grpcclient import tritonclient.grpc as grpcclient
import numpy as np from PIL import Image
from PIL import Image, ImageOps
from minio import Minio
from tritonclient.utils import np_to_triton_dtype from tritonclient.utils import np_to_triton_dtype
from app.core.config import * from app.core.config import *
@@ -40,7 +39,7 @@ class GenerateRelightImage:
self.prompt = request_data.prompt self.prompt = request_data.prompt
self.seed = "1" self.seed = "1"
self.negative_prompt = 'lowres, bad anatomy, bad hands, cropped, worst quality' self.negative_prompt = 'lowres, bad anatomy, bad hands, cropped, worst quality'
self.direction = "Right Light" self.direction = request_data.direction
self.image_url = request_data.image_url self.image_url = request_data.image_url
self.image = oss_get_image(bucket=self.image_url.split('/')[0], object_name=self.image_url[self.image_url.find('/') + 1:], data_type="cv2") self.image = oss_get_image(bucket=self.image_url.split('/')[0], object_name=self.image_url[self.image_url.find('/') + 1:], data_type="cv2")
self.tasks_id = request_data.tasks_id self.tasks_id = request_data.tasks_id