305 lines
11 KiB
Python
Executable File
305 lines
11 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
测试脚本：专门测试图像处理功能
|
||
不依赖于AI接口，仅测试图像拼贴和海报生成功能
|
||
"""
|
||
import os
|
||
import sys
|
||
import json
|
||
import argparse
|
||
import logging
|
||
import random
|
||
import numpy as np
|
||
from datetime import datetime
|
||
|
||
# 添加项目根目录到Python路径
|
||
PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
if PROJECT_ROOT not in sys.path:
|
||
sys.path.append(PROJECT_ROOT)
|
||
|
||
# 导入所需的图像处理模块
|
||
try:
|
||
import core.simple_collage as simple_collage
|
||
import core.poster_gen as poster_gen
|
||
from utils.resource_loader import ResourceLoader
|
||
except ImportError as e:
|
||
logging.critical(f"导入模块失败: {e}")
|
||
logging.critical(f"确保'{PROJECT_ROOT}'在sys.path中，并且所有依赖已安装")
|
||
sys.exit(1)
|
||
|
||
def load_config(config_path="poster_gen_config.json"):
|
||
"""加载配置文件"""
|
||
try:
|
||
with open(config_path, 'r', encoding='utf-8') as f:
|
||
config = json.load(f)
|
||
logging.info(f"从 {config_path} 加载配置成功")
|
||
return config
|
||
except FileNotFoundError:
|
||
logging.error(f"配置文件未找到: {config_path}")
|
||
return None
|
||
except json.JSONDecodeError:
|
||
logging.error(f"配置文件解析失败: {config_path}，请检查JSON格式")
|
||
return None
|
||
except Exception as e:
|
||
logging.exception(f"加载配置文件时发生错误:")
|
||
return None
|
||
|
||
def create_test_images(output_dir, count=5, size=(800, 600)):
|
||
"""创建测试用的彩色图像"""
|
||
os.makedirs(output_dir, exist_ok=True)
|
||
logging.info(f"创建 {count} 张测试图像到 {output_dir}")
|
||
|
||
image_paths = []
|
||
for i in range(count):
|
||
# 创建随机彩色图像
|
||
img = np.zeros((size[1], size[0], 3), dtype=np.uint8)
|
||
color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
|
||
img[:] = color
|
||
|
||
# 保存图像
|
||
img_path = os.path.join(output_dir, f"test_img_{i+1}.jpg")
|
||
try:
|
||
import cv2
|
||
cv2.imwrite(img_path, img)
|
||
image_paths.append(img_path)
|
||
logging.info(f"创建测试图像: {img_path}")
|
||
except Exception as e:
|
||
logging.error(f"创建测试图像失败: {e}")
|
||
|
||
return image_paths
|
||
|
||
def test_collage(config, output_dir, test_images_dir=None, collage_count=3):
|
||
"""测试图像拼贴功能"""
|
||
logging.info("=== 开始测试图像拼贴功能 ===")
|
||
|
||
# 准备输出目录
|
||
collage_output_dir = os.path.join(output_dir, "collages")
|
||
os.makedirs(collage_output_dir, exist_ok=True)
|
||
|
||
# 准备测试图像
|
||
if test_images_dir is None or not os.path.exists(test_images_dir):
|
||
test_images_dir = os.path.join(output_dir, "test_images")
|
||
image_paths = create_test_images(test_images_dir)
|
||
else:
|
||
image_paths = [os.path.join(test_images_dir, f) for f in os.listdir(test_images_dir)
|
||
if f.lower().endswith(('.jpg', '.jpeg', '.png'))]
|
||
|
||
if not image_paths:
|
||
logging.error("无有效测试图像可用")
|
||
return False
|
||
|
||
# 测试不同风格的拼贴
|
||
styles = ["standard", "asymmetric", "filmstrip", "overlap", "mosaic"]
|
||
success_count = 0
|
||
|
||
for i in range(collage_count):
|
||
style = styles[i % len(styles)]
|
||
try:
|
||
logging.info(f"生成风格为 '{style}' 的拼贴图 {i+1}/{collage_count}")
|
||
output_path = os.path.join(collage_output_dir, f"collage_{style}_{i+1}.jpg")
|
||
|
||
# 创建拼贴图
|
||
collage_creator = simple_collage.ImageCollageCreator(
|
||
target_size=(900, 900),
|
||
style=style
|
||
)
|
||
|
||
# 随机选择3-5张图片
|
||
num_images = random.randint(3, min(5, len(image_paths)))
|
||
selected_images = random.sample(image_paths, num_images)
|
||
|
||
# 加载图像
|
||
images = []
|
||
for img_path in selected_images:
|
||
try:
|
||
import cv2
|
||
img = cv2.imread(img_path)
|
||
if img is not None:
|
||
images.append(img)
|
||
except Exception as e:
|
||
logging.warning(f"加载图像 {img_path} 失败: {e}")
|
||
|
||
if not images:
|
||
logging.warning(f"无法加载任何图像，跳过拼贴 {i+1}")
|
||
continue
|
||
|
||
# 创建拼贴
|
||
result = collage_creator.create_collage(images)
|
||
|
||
# 保存结果
|
||
cv2.imwrite(output_path, result)
|
||
logging.info(f"拼贴图保存到: {output_path}")
|
||
success_count += 1
|
||
|
||
except Exception as e:
|
||
logging.exception(f"创建拼贴图 {i+1} 时出错:")
|
||
|
||
logging.info(f"拼贴测试完成: 成功 {success_count}/{collage_count}")
|
||
return success_count > 0
|
||
|
||
def test_poster_generation(config, output_dir, collage_dir=None):
|
||
"""测试海报生成功能"""
|
||
logging.info("=== 开始测试海报生成功能 ===")
|
||
|
||
# 准备输出目录
|
||
poster_output_dir = os.path.join(output_dir, "posters")
|
||
os.makedirs(poster_output_dir, exist_ok=True)
|
||
|
||
# 获取海报素材目录
|
||
poster_assets_dir = config.get("poster_assets_base_dir")
|
||
if not poster_assets_dir or not os.path.exists(poster_assets_dir):
|
||
logging.error(f"海报素材目录不存在或未配置: {poster_assets_dir}")
|
||
return False
|
||
|
||
# 获取或创建底图
|
||
base_images = []
|
||
|
||
# 使用之前创建的拼贴图
|
||
if collage_dir and os.path.exists(collage_dir):
|
||
collage_files = [f for f in os.listdir(collage_dir) if f.lower().endswith('.jpg')]
|
||
for file in collage_files:
|
||
base_images.append(os.path.join(collage_dir, file))
|
||
|
||
# 如果没有拼贴图或数量不足，创建纯色测试图像
|
||
if len(base_images) < 3:
|
||
test_img_dir = os.path.join(output_dir, "base_images")
|
||
new_images = create_test_images(test_img_dir, count=3, size=(900, 1200))
|
||
base_images.extend(new_images)
|
||
|
||
# 准备一些测试文本
|
||
test_texts = [
|
||
{
|
||
"title": "云南大理古城悠闲一日游",
|
||
"subtitle": "探寻苍山洱海间的千年历史",
|
||
"content": "大理古城，一座被苍山洱海环抱的千年古城，漫步其中，感受白族建筑与现代艺术的完美融合。"
|
||
},
|
||
{
|
||
"title": "秋日北京香山红叶之旅",
|
||
"subtitle": "首都最美的秋天风景线",
|
||
"content": "每年深秋，香山红叶如火如荼，是摄影爱好者的天堂，也是周末休闲的绝佳去处。"
|
||
},
|
||
{
|
||
"title": "青岛海滨度假攻略",
|
||
"subtitle": "啤酒与海鲜的夏日盛宴",
|
||
"content": '青岛，被誉为"东方瑞士"，在这里，你可以感受到欧式建筑与中国沿海城市的独特魅力。'
|
||
}
|
||
]
|
||
|
||
# 创建海报生成器
|
||
try:
|
||
poster_generator = poster_gen.PosterGenerator(
|
||
poster_save_dir=poster_output_dir,
|
||
assets_base_dir=poster_assets_dir,
|
||
poster_size=tuple(config.get("poster_target_size", [900, 1200]))
|
||
)
|
||
logging.info("海报生成器初始化成功")
|
||
except Exception as e:
|
||
logging.exception("初始化海报生成器失败:")
|
||
return False
|
||
|
||
# 生成多张测试海报
|
||
success_count = 0
|
||
for i, base_img_path in enumerate(base_images[:3]):
|
||
try:
|
||
text_data = test_texts[i % len(test_texts)]
|
||
output_path = os.path.join(poster_output_dir, f"poster_{i+1}.jpg")
|
||
|
||
logging.info(f"生成海报 {i+1}/3，使用底图: {os.path.basename(base_img_path)}")
|
||
|
||
# 调用海报生成
|
||
poster_generator.create_poster(
|
||
base_image_path=base_img_path,
|
||
poster_text_config={
|
||
"title": text_data["title"],
|
||
"subtitle": text_data["subtitle"],
|
||
"content": text_data["content"],
|
||
"has_second_paragraph": random.random() < 0.5
|
||
},
|
||
output_path=output_path
|
||
)
|
||
|
||
if os.path.exists(output_path):
|
||
logging.info(f"海报已保存到: {output_path}")
|
||
success_count += 1
|
||
else:
|
||
logging.warning(f"海报创建似乎成功，但文件未找到: {output_path}")
|
||
|
||
except Exception as e:
|
||
logging.exception(f"创建海报 {i+1} 时出错:")
|
||
|
||
logging.info(f"海报测试完成: 成功 {success_count}/3")
|
||
return success_count > 0
|
||
|
||
def main():
|
||
# 设置日志
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format='%(asctime)s - %(levelname)s - [%(filename)s:%(lineno)d] - %(message)s',
|
||
datefmt='%Y-%m-%d %H:%M:%S'
|
||
)
|
||
|
||
# 命令行参数
|
||
parser = argparse.ArgumentParser(description='测试图像处理功能')
|
||
parser.add_argument('--config', type=str, default='poster_gen_config.json', help='配置文件路径')
|
||
parser.add_argument('--output', type=str, default=None, help='输出目录，默认为result/test_image_TIMESTAMP')
|
||
parser.add_argument('--collage_only', action='store_true', help='仅测试拼贴功能')
|
||
parser.add_argument('--poster_only', action='store_true', help='仅测试海报生成功能')
|
||
args = parser.parse_args()
|
||
|
||
# 加载配置
|
||
logging.info(f"加载配置: {args.config}")
|
||
config = load_config(args.config)
|
||
if not config:
|
||
logging.critical("配置加载失败，无法继续测试")
|
||
sys.exit(1)
|
||
|
||
# 准备输出目录
|
||
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
|
||
if args.output:
|
||
output_base_dir = args.output
|
||
else:
|
||
output_base_dir = os.path.join(config.get("output_dir", "result"), f"test_image_{timestamp}")
|
||
|
||
os.makedirs(output_base_dir, exist_ok=True)
|
||
logging.info(f"测试输出将保存到: {output_base_dir}")
|
||
|
||
# 执行测试
|
||
test_results = []
|
||
|
||
# 如果两个only标志都没设置，则都测试
|
||
test_all = not args.collage_only and not args.poster_only
|
||
|
||
# 测试拼贴
|
||
if args.collage_only or test_all:
|
||
collage_success = test_collage(config, output_base_dir)
|
||
test_results.append(("拼贴功能", collage_success))
|
||
|
||
# 如果只测试拼贴或需要连续测试，且拼贴成功了
|
||
if collage_success and (args.collage_only or test_all):
|
||
collage_dir = os.path.join(output_base_dir, "collages")
|
||
else:
|
||
collage_dir = None
|
||
else:
|
||
collage_dir = None
|
||
|
||
# 测试海报生成
|
||
if args.poster_only or test_all:
|
||
poster_success = test_poster_generation(config, output_base_dir, collage_dir)
|
||
test_results.append(("海报生成功能", poster_success))
|
||
|
||
# 输出结果摘要
|
||
logging.info("\n=== 测试结果摘要 ===")
|
||
all_success = True
|
||
for name, result in test_results:
|
||
status = "成功" if result else "失败"
|
||
logging.info(f"{name}: {status}")
|
||
all_success = all_success and result
|
||
|
||
logging.info(f"\n测试{'全部通过' if all_success else '部分失败'}!")
|
||
logging.info(f"所有测试输出保存在: {output_base_dir}")
|
||
|
||
sys.exit(0 if all_success else 1)
|
||
|
||
if __name__ == "__main__":
|
||
main() |