feat image2sketch 变更模型

fix
This commit is contained in:
zhouchengrong
2024-09-30 10:57:12 +08:00
parent 19fa165a1e
commit 85942167f3
6 changed files with 200 additions and 7 deletions

View File

@@ -6,7 +6,7 @@ from fastapi import APIRouter, HTTPException
from app.schemas.image2sketch import Image2SketchModel
from app.schemas.response_template import ResponseModel
from app.service.image2sketch.server import Image2SketchServer
from app.service.image2sketch_2.server import processing_pipeline
router = APIRouter()
logger = logging.getLogger()
@@ -25,8 +25,7 @@ def image2sketch(request_item: Image2SketchModel):
示例参数:
{
"image_url": "test/image2sketch/real_Dress_3200fecdc83d0c556c2bd96aedbd7fbf.jpg_Img.jpg",
"style_image_url": "test/image2sketch/style_3.png",
"default_style": "1",
"default_style": 0,
"sketch_bucket": "test",
"sketch_name": "image2sketch/test.png"
}
@@ -34,8 +33,12 @@ def image2sketch(request_item: Image2SketchModel):
try:
start_time = time.time()
logger.info(f"image2sketch request item is : @@@@@@:{json.dumps(request_item.dict())}")
service = Image2SketchServer(request_item)
sketch_url = service.get_result()
sketch_url = processing_pipeline(
image_url=request_item.image_url,
thickness=request_item.default_style,
sketch_bucket=request_item.sketch_bucket,
sketch_name=request_item.sketch_name
)
logger.info(f"run time is : {time.time() - start_time}")
except Exception as e:
logger.warning(f"image2sketch Run Exception @@@@@@:{e}")