feat image2sketch 变更模型
fix
This commit is contained in:
@@ -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}")
|
||||
|
||||
Reference in New Issue
Block a user