185 lines
6.5 KiB
Python
185 lines
6.5 KiB
Python
#!/usr/bin/env python3
|
||
"""CHAR-004 R2 候选质检 · Qwen-VL · VA-05-001-R2标准"""
|
||
import sys, os, json, base64, subprocess
|
||
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
from secrets_loader import secret, endpoint
|
||
|
||
API_KEY = secret("SC-004")
|
||
EP = endpoint("EPT-002")
|
||
FALLBACK_EP = "https://dashscope.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation"
|
||
|
||
CANDIDATES_DIR = os.path.expanduser("~/guanghulab/video-ai-system/assets/candidates/D157-VA-05-001-R2/CHAR-004-ZhugeFeng")
|
||
CANDIDATES = [f"{CANDIDATES_DIR}/CHAR-004-R2-candidate-{i:02d}-1080x1920.jpg" for i in range(1, 5)]
|
||
|
||
QC_PROMPT = """你是动画资产质量检查员。检查这张CHAR-004诸葛风立绘候选图，按以下标准逐项打分。
|
||
|
||
## R2规范必查项
|
||
|
||
1. 服装完整性（一票否决）：
|
||
- 裤子膝盖是否有破洞/破损？是/否
|
||
- 衣服下摆是否有毛边/破损/撕裂？是/否
|
||
- 鞋子是否有泥污/脏污？是/否
|
||
- 服装是否有污渍？是/否
|
||
- 是否有补丁？是/否
|
||
- 整体是否"乞丐化"（像乞丐而不是清贫少年）？
|
||
|
||
2. 服装部件准确性：
|
||
- 上衣是否为灰蓝色交领短打、右衽？是/否
|
||
- 是否系深棕色粗麻绳腰带？是/否
|
||
- 裤子是否深灰蓝色、裤脚塞进靴子？是/否
|
||
- 鞋子是否黑色粗布短靴？是/否
|
||
- 腰间是否有旧布包？是/否
|
||
- 有无不该有的配饰（玉佩、香囊等）？是/否
|
||
|
||
3. 面部质量：
|
||
- 五官是否清晰无崩坏？是/否
|
||
- 是否少年感强、气质偏冷硬？是/否
|
||
- 有没有和苏白（温润仙气型）相似？是/否
|
||
|
||
4. 整体：
|
||
- 人物肢体比例是否正常？是/否
|
||
- 画面是否清晰无噪点？是/否
|
||
- 是否全身从头到鞋完整展示？是/否
|
||
|
||
输出严格JSON，不要额外文字：
|
||
{
|
||
"candidate": N,
|
||
"pass": true/false,
|
||
"score": 0-100,
|
||
"hole_in_pants": true/false,
|
||
"frayed_hem": true/false,
|
||
"dirty_shoes": true/false,
|
||
"stains": true/false,
|
||
"patches": true/false,
|
||
"beggar_look": true/false,
|
||
"correct_top": true/false,
|
||
"correct_belt": true/false,
|
||
"correct_pants": true/false,
|
||
"correct_shoes": true/false,
|
||
"has_bag": true/false,
|
||
"extra_accessories": true/false,
|
||
"face_intact": true/false,
|
||
"cold_temperament": true/false,
|
||
"similar_to_su_bai": true/false,
|
||
"anatomy_ok": true/false,
|
||
"clear_image": true/false,
|
||
"full_body": true/false,
|
||
"summary": "一句话总结"
|
||
}"""
|
||
|
||
def encode_image(path):
|
||
with open(path, "rb") as f:
|
||
b64 = base64.b64encode(f.read()).decode()
|
||
ext = path.rsplit(".", 1)[-1].lower()
|
||
mime = {"jpg": "jpeg", "jpeg": "jpeg", "png": "png"}.get(ext, "jpeg")
|
||
return f"data:image/{mime};base64,{b64}"
|
||
|
||
def call_qwen(image_b64):
|
||
body = json.dumps({
|
||
"model": "qwen-vl-max",
|
||
"input": {
|
||
"messages": [{
|
||
"role": "user",
|
||
"content": [
|
||
{"image": image_b64},
|
||
{"text": QC_PROMPT}
|
||
]
|
||
}]
|
||
}
|
||
})
|
||
|
||
endpoints = [EP, FALLBACK_EP]
|
||
for ep in endpoints:
|
||
try:
|
||
proc = subprocess.run(
|
||
["curl", "-s", "-m", "60", "--noproxy", "*",
|
||
"-X", "POST", ep,
|
||
"-H", f"Authorization: Bearer {API_KEY}",
|
||
"-H", "Content-Type: application/json",
|
||
"--data-binary", "@-"],
|
||
input=body, capture_output=True, text=True, timeout=65
|
||
)
|
||
if proc.returncode != 0 or not proc.stdout.strip():
|
||
continue
|
||
resp = json.loads(proc.stdout)
|
||
if "output" in resp:
|
||
content = resp["output"]["choices"][0]["message"]["content"][0]["text"]
|
||
# Clean markdown wrappers
|
||
if "```" in content:
|
||
content = content.split("```")[1]
|
||
if content.startswith("json"):
|
||
content = content[4:]
|
||
content = content.split("```")[0]
|
||
return json.loads(content.strip())
|
||
elif "error" in resp:
|
||
print(f" [Qwen error] {resp['error']}", file=sys.stderr)
|
||
continue
|
||
except Exception as e:
|
||
print(f" [fail {ep}] {e}", file=sys.stderr)
|
||
continue
|
||
return None
|
||
|
||
def main():
|
||
print("=" * 60)
|
||
print("CHAR-004 R2 候选质检 · Qwen-VL · VA-05-001-R2标准")
|
||
print("=" * 60)
|
||
|
||
results = []
|
||
for i, path in enumerate(CANDIDATES):
|
||
print(f"\n[{i+1}/4] 质检候选 {i+1}...")
|
||
img = encode_image(path)
|
||
qc = call_qwen(img)
|
||
|
||
if qc:
|
||
print(f" score={qc.get('score')} pass={qc.get('pass')}")
|
||
print(f" 破洞={qc.get('hole_in_pants')} 毛边={qc.get('frayed_hem')} 脏鞋={qc.get('dirty_shoes')}")
|
||
print(f" 乞丐化={qc.get('beggar_look')} 和苏白像={qc.get('similar_to_su_bai')}")
|
||
print(f" {qc.get('summary')}")
|
||
results.append(qc)
|
||
else:
|
||
print(f" ✗ Qwen-VL调用失败")
|
||
results.append({"candidate": i+1, "pass": False, "error": "Qwen-VL call failed"})
|
||
|
||
# 汇总
|
||
print("\n" + "=" * 60)
|
||
print("质检汇总")
|
||
print("=" * 60)
|
||
|
||
passed = sum(1 for r in results if r.get("pass"))
|
||
print(f"通过: {passed}/4")
|
||
|
||
for r in results:
|
||
if not r.get("pass"):
|
||
issues = []
|
||
if r.get("hole_in_pants"): issues.append("裤子破洞")
|
||
if r.get("frayed_hem"): issues.append("下摆毛边")
|
||
if r.get("dirty_shoes"): issues.append("鞋子脏污")
|
||
if r.get("stains"): issues.append("污渍")
|
||
if r.get("patches"): issues.append("补丁")
|
||
if r.get("beggar_look"): issues.append("乞丐化")
|
||
if r.get("extra_accessories"): issues.append("多余配饰")
|
||
print(f" 候选{r.get('candidate')} ✗: {', '.join(issues) if issues else 'other'}")
|
||
|
||
# 写入JSON
|
||
out_path = os.path.join(CANDIDATES_DIR, "QC-R2-BATCH.json")
|
||
report = {
|
||
"spec": "VA-05-001-R2",
|
||
"asset_id": "CHAR-004-ZhugeFeng",
|
||
"qc_tool": "Qwen-VL (qwen-vl-max)",
|
||
"results": results,
|
||
"summary": {
|
||
"passed": passed,
|
||
"total": 4,
|
||
"verdict": "APPROVED" if passed >= 4 else "REJECTED"
|
||
}
|
||
}
|
||
with open(out_path, "w") as f:
|
||
json.dump(report, f, ensure_ascii=False, indent=2)
|
||
print(f"\nQC报告已保存: {out_path}")
|
||
|
||
return 0 if passed >= 4 else 1
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|