feat(新功能): clothing seg

fix(修复bug):
docs(文档变更):
refactor(重构):
test(增加测试):
This commit is contained in:
zchengrong
2025-04-11 17:14:59 +08:00
parent 3593f0d431
commit f83a202b20
4 changed files with 217 additions and 2 deletions

View File

@@ -0,0 +1,51 @@
import json
import logging
from fastapi import APIRouter, HTTPException
from app.schemas.response_template import ResponseModel
from app.schemas.clothing_seg import ClothingSegModel
from app.service.clothing_seg.service import ClothingSeg
router = APIRouter()
logger = logging.getLogger()
@router.post("/clothing_seg")
def clothing_seg(request_item: ClothingSegModel):
"""
创建一个具有以下参数的请求体:
- **user_id**: 用户id
- **image_data**: 图片数据
{
"image_url": "test/clothing_seg/dress.jpg",
"image_type": "product"
}
示例参数:
{
"user_id": 89,
"image_data": [
{
"image_url": "test/clothing_seg/dress.jpg",
"image_type": "sketch"
},
{
"image_url": "test/clothing_seg/skirt_559.jpg",
"image_type": "sketch"
},
{
"image_url": "test/clothing_seg/10144613.jpg",
"image_type": "product"
}
]
}
"""
try:
logger.info(f"clothing_seg request item is : @@@@@@:{json.dumps(request_item.dict())}")
server = ClothingSeg(request_item)
result_url = server.get_result()
except Exception as e:
logger.warning(f"clothing_seg Run Exception @@@@@@:{e}")
raise HTTPException(status_code=404, detail=str(e))
return ResponseModel(data=result_url)

View File

@@ -1,6 +1,6 @@
from fastapi import APIRouter
from app.api import api_agent_generate_image, api_recommendation
from app.api import api_agent_generate_image
from app.api import api_attribute_retrieve, api_query_image
from app.api import api_brand_dna
from app.api import api_brighten
@@ -12,6 +12,7 @@ from app.api import api_image2sketch
from app.api import api_mannequins_edit
from app.api import api_pose_transform
from app.api import api_prompt_generation
from app.api import api_clothing_seg
from app.api import api_super_resolution
from app.api import api_test
@@ -29,7 +30,8 @@ router.include_router(api_image2sketch.router, tags=['api_image2sketch'], prefix
router.include_router(api_brighten.router, tags=['api_brighten'], prefix="/api")
router.include_router(api_query_image.router, tags=['api_query_image'], prefix="/api")
router.include_router(api_brand_dna.router, tags=['api_brand_dna'], prefix="/api")
router.include_router(api_recommendation.router, tags=['api_recommendation'], prefix="/api")
# router.include_router(api_recommendation.router, tags=['api_recommendation'], prefix="/api")
router.include_router(api_mannequins_edit.router, tags=['api_mannequins_edit'], prefix="/api")
router.include_router(api_agent_generate_image.router, tags=['api_agent_generate_image'], prefix="/api")
router.include_router(api_pose_transform.router, tags=['api_pose_transform'], prefix="/api")
router.include_router(api_clothing_seg.router, tags=['api_clothing_seg'], prefix="/api")