更新产品配置和技能定义
This commit is contained in:
64
.codebuddy/skills/image-tools/scripts/remove-background.py
Normal file
64
.codebuddy/skills/image-tools/scripts/remove-background.py
Normal file
@@ -0,0 +1,64 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
产品图片抠图工具
|
||||
使用rembg库自动去除图片背景
|
||||
"""
|
||||
import os
|
||||
import sys
|
||||
|
||||
try:
|
||||
from rembg import remove
|
||||
from PIL import Image
|
||||
except ImportError:
|
||||
print("Missing dependencies, installing...")
|
||||
os.system("pip install rembg pillow")
|
||||
from rembg import remove
|
||||
from PIL import Image
|
||||
|
||||
def remove_background(input_path, output_path):
|
||||
"""
|
||||
去除图片背景
|
||||
"""
|
||||
try:
|
||||
print(f"Processing: {input_path}")
|
||||
|
||||
# 打开原始图片
|
||||
with open(input_path, 'rb') as input_file:
|
||||
input_data = input_file.read()
|
||||
|
||||
# 去除背景
|
||||
output_data = remove(input_data)
|
||||
|
||||
# 保存结果
|
||||
with open(output_path, 'wb') as output_file:
|
||||
output_file.write(output_data)
|
||||
|
||||
print(f"Saved: {output_path}")
|
||||
return True
|
||||
except Exception as e:
|
||||
print(f"Error: {e}")
|
||||
return False
|
||||
|
||||
if __name__ == "__main__":
|
||||
# 产品图片路径
|
||||
image_dir = r"g:\stonelan-workspace\membank\产品\飞牛 NAS\fyd-亚克力机箱 - 单\image"
|
||||
|
||||
# 要处理的图片
|
||||
input_image = os.path.join(image_dir, "正面.jpg")
|
||||
output_image = os.path.join(image_dir, "正面_抠图.png")
|
||||
|
||||
# 检查文件是否存在
|
||||
if not os.path.exists(input_image):
|
||||
print(f"File not found: {input_image}")
|
||||
sys.exit(1)
|
||||
|
||||
# 执行抠图
|
||||
success = remove_background(input_image, output_image)
|
||||
|
||||
if success:
|
||||
print(f"\nBackground removal completed!")
|
||||
print(f"Original: {input_image}")
|
||||
print(f"Result: {output_image}")
|
||||
else:
|
||||
print("\nBackground removal failed!")
|
||||
sys.exit(1)
|
||||
58
.codebuddy/skills/image-tools/scripts/render-main-image.js
Normal file
58
.codebuddy/skills/image-tools/scripts/render-main-image.js
Normal file
@@ -0,0 +1,58 @@
|
||||
const puppeteer = require('puppeteer');
|
||||
const path = require('path');
|
||||
const fs = require('fs');
|
||||
|
||||
async function convertMainImageToPng() {
|
||||
console.log('🎨 启动主图渲染工具...\n');
|
||||
|
||||
const browser = await puppeteer.launch({
|
||||
headless: 'new',
|
||||
args: ['--no-sandbox', '--disable-setuid-sandbox']
|
||||
});
|
||||
|
||||
const page = await browser.newPage();
|
||||
|
||||
await page.setViewport({ width: 800, height: 800 });
|
||||
|
||||
const mainImageDir = 'G:\\stonelan-workspace\\membank\\产品\\飞牛 NAS\\fyd-亚克力机箱 - 单\\主图';
|
||||
|
||||
const htmlFiles = fs.readdirSync(mainImageDir)
|
||||
.filter(file => file.endsWith('.html'));
|
||||
|
||||
console.log(`📁 找到 ${htmlFiles.length} 个主图HTML文件需要渲染\n`);
|
||||
|
||||
for (const htmlFile of htmlFiles) {
|
||||
try {
|
||||
const htmlPath = path.join(mainImageDir, htmlFile);
|
||||
const pngFile = htmlFile.replace('.html', '.png');
|
||||
const pngPath = path.join(mainImageDir, pngFile);
|
||||
|
||||
console.log(`📄 正在渲染: ${htmlFile}`);
|
||||
|
||||
const fileUrl = `file:///${htmlPath.replace(/\\/g, '/')}`;
|
||||
await page.goto(fileUrl, {
|
||||
waitUntil: 'networkidle0',
|
||||
timeout: 10000
|
||||
});
|
||||
|
||||
await page.screenshot({
|
||||
path: pngPath,
|
||||
fullPage: true,
|
||||
type: 'png'
|
||||
});
|
||||
|
||||
console.log(`✅ 已生成: ${pngFile}\n`);
|
||||
} catch (error) {
|
||||
console.error(`❌ 渲染失败 ${htmlFile}: ${error.message}\n`);
|
||||
}
|
||||
}
|
||||
|
||||
await browser.close();
|
||||
|
||||
console.log('✨ 所有主图渲染完成!');
|
||||
}
|
||||
|
||||
convertMainImageToPng().catch(error => {
|
||||
console.error('💥 渲染过程出错:', error);
|
||||
process.exit(1);
|
||||
});
|
||||
Reference in New Issue
Block a user