feat(新功能): batch generate 增加接口说明,design batch队列修改
fix(修复bug): docs(文档变更): refactor(重构): test(增加测试):
This commit is contained in:
28
app/service/generate_batch_image/tasks.py
Normal file
28
app/service/generate_batch_image/tasks.py
Normal file
@@ -0,0 +1,28 @@
|
|||||||
|
# import logging
|
||||||
|
#
|
||||||
|
# from celery import Celery
|
||||||
|
#
|
||||||
|
# from app.service.generate_batch_image.service_batch_generate_product_image import batch_generate_product
|
||||||
|
# from app.service.generate_batch_image.service_batch_generate_relight_image import batch_generate_relight
|
||||||
|
# from app.service.generate_batch_image.service_batch_pose_transform import batch_generate_pose_transform
|
||||||
|
#
|
||||||
|
# logger = logging.getLogger()
|
||||||
|
# celery_app = Celery('tasks', broker=f'amqp://rabbit:123456@18.167.251.121:5672//', backend='rpc://', BROKER_CONNECTION_RETRY_ON_STARTUP=True)
|
||||||
|
# celery_app.conf.worker_log_format = '%(asctime)s %(filename)s [line:%(lineno)d] %(levelname)s %(message)s'
|
||||||
|
# celery_app.conf.worker_hijack_root_logger = False
|
||||||
|
# logging.getLogger('pika').setLevel(logging.WARNING)
|
||||||
|
#
|
||||||
|
#
|
||||||
|
# @celery_app.task
|
||||||
|
# def batch_pose_transform_tasks(batch_request_data):
|
||||||
|
# batch_generate_pose_transform(batch_request_data)
|
||||||
|
#
|
||||||
|
#
|
||||||
|
# @celery_app.task
|
||||||
|
# def batch_generate_relight_tasks(batch_request_data):
|
||||||
|
# batch_generate_relight(batch_request_data)
|
||||||
|
#
|
||||||
|
#
|
||||||
|
# @celery_app.task
|
||||||
|
# def batch_generate_product_tasks(batch_request_data):
|
||||||
|
# batch_generate_product(batch_request_data)
|
||||||
36
app/service/generate_batch_image/test.py
Normal file
36
app/service/generate_batch_image/test.py
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
from app.schemas.generate_image import BatchGenerateRelightImageModel, BatchGenerateProductImageModel
|
||||||
|
from app.service.generate_batch_image.service_batch_generate_product_image import batch_generate_product
|
||||||
|
|
||||||
|
from app.service.generate_batch_image.service_batch_generate_relight_image import batch_generate_relight
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
rd = BatchGenerateProductImageModel(
|
||||||
|
tasks_id="test1-89",
|
||||||
|
image_strength=0.7,
|
||||||
|
prompt=" The best quality, masterpiece, real image.Outwear,high quality clothing details,8K realistic,HDR",
|
||||||
|
image_url="aida-results/result_40b1a2fe-e220-11ef-9bfa-0242ac150003.png",
|
||||||
|
product_type="single",
|
||||||
|
batch_size=2
|
||||||
|
)
|
||||||
|
x = batch_generate_product.delay(rd.dict())
|
||||||
|
print(x)
|
||||||
|
|
||||||
|
"""relight"""
|
||||||
|
# rd = BatchGenerateRelightImageModel(
|
||||||
|
# tasks_id="123-89",
|
||||||
|
# # prompt="beautiful woman, detailed face, sunshine, outdoor, warm atmosphere",
|
||||||
|
# prompt="Colorful black",
|
||||||
|
# image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
|
||||||
|
# direction="Right Light",
|
||||||
|
# product_type="single",
|
||||||
|
# batch_size=2
|
||||||
|
# )
|
||||||
|
# batch_generate_relight.delay(rd.dict())
|
||||||
|
"""pose transform"""
|
||||||
|
# rd = BatchPoseTransformModel(
|
||||||
|
# tasks_id="123-89",
|
||||||
|
# image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
|
||||||
|
# pose_id="1",
|
||||||
|
# batch_size=10
|
||||||
|
# )
|
||||||
|
# batch_pose_transform_tasks.delay(rd.dict())
|
||||||
Reference in New Issue
Block a user