Mrpzx
977f97995a
覆盖文生 3D、图生 3D、多视图、批量生成，以及重拓扑、绑定、动画、智能分割、 纹理、PBR、智能 UV、部件补全、格式导出与下载。 代码组织：一个功能一个模块（tripo/features/），入口薄壳 tripocli.py。
77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
"""批量图生 3D — batch image to model.
|
||
|
||
tripocli batch -i a.png,b.png,c.png --wait -o ./out
|
||
|
||
Verified against the live API: the batch endpoint takes ``image`` as an *array*
|
||
(the field is singular, not ``images``) and answers with an array of
|
||
``{project_id, operator_id}`` — one entry per submitted image.
|
||
"""
|
||
|
||
import os
|
||
|
||
from . import _common
|
||
|
||
NAME = "batch"
|
||
HELP = "批量图生 3D（多张图片一次提交，返回多个任务）"
|
||
|
||
|
||
def add_arguments(parser):
|
||
parser.add_argument("-i", "--images", required=True, help="逗号分隔的本地图片路径")
|
||
_common.add_generation_options(parser)
|
||
_common.add_wait_options(parser, "输出目录（每个结果保存为 <project_id>.glb）")
|
||
|
||
|
||
def run(ctx):
|
||
args = ctx.args
|
||
paths = [p.strip() for p in args.images.split(",") if p.strip()]
|
||
if not paths:
|
||
raise ValueError("--images 至少需要一张图片")
|
||
|
||
images = []
|
||
for path in paths:
|
||
ctx.out.line(f"📤 上传并审核图片: {path}")
|
||
images.append(ctx.client.upload_image(path))
|
||
|
||
body = _common.generation_body(args)
|
||
results = ctx.client.get_data(
|
||
"POST",
|
||
"/v2/studio/operation/batch_image_to_model",
|
||
body={
|
||
"image": images,
|
||
"model_version": body["model_version"],
|
||
"face_limit": body["face_limit"],
|
||
"quad": body["quad"],
|
||
"texture": body["texture"],
|
||
"visibility": body["visibility"],
|
||
},
|
||
)
|
||
results = results or []
|
||
|
||
if not args.wait:
|
||
ctx.out.data(
|
||
results,
|
||
human=lambda: (
|
||
ctx.out.line(f"✅ 已提交 {len(results)} 个任务"),
|
||
*[ctx.out.line(f" project {r.get('project_id')} operator {r.get('operator_id')}") for r in results],
|
||
),
|
||
)
|
||
return results
|
||
|
||
out_dir = args.output or "."
|
||
payload = []
|
||
for item in results:
|
||
project_id = item.get("project_id")
|
||
wait_task = _common.wait_and_report(ctx, item.get("operator_id"))
|
||
model_url, _detail = _common.fetch_model_url(ctx, project_id)
|
||
target = None
|
||
if model_url:
|
||
target = os.path.abspath(os.path.join(out_dir, f"{project_id}.glb"))
|
||
ctx.client.download(model_url, target)
|
||
payload.append({"task": wait_task, "project_id": project_id, "model_url": model_url, "output_file": target})
|
||
|
||
ctx.out.data(
|
||
payload,
|
||
human=lambda: [ctx.out.line(f"✨ {p['output_file'] or p['project_id']}") for p in payload],
|
||
)
|
||
return payload
|