Update egg-api.js

egg-api.js

@@ -7,7 +7,8 @@ const axios = require('axios');
 const sharp = require('sharp');
 
 const app = express();
-const port = 7860; // 将端口改为 7860
+const port = 7860;
+
 // 从文件夹加载所有分类器
 async function loadClassifiersFromFolder(folderPath) {
   const classifiers = [];
@@ -42,7 +43,7 @@ async function downloadAndConvertImage(url) {
 }
 
 // 图像处理函数
-async function processImage(imageBuffer, classifiersFolder, replacementImagePath) {
+async function processImage(imageBuffer, classifiersFolder, replacementImageBuffer) {
   try {
     const classifiers = await loadClassifiersFromFolder(classifiersFolder);
 
@@ -64,22 +65,29 @@ async function processImage(imageBuffer, classifiersFolder, replacementImagePath
       allFaces.push(...faces);
     });
 
-    const replacementFace = await Jimp.read(replacementImagePath);
+    const replacementFace = await Jimp.read(replacementImageBuffer);
     allFaces.forEach((faceRect, i) => {
-      const resizedReplacementFace = replacementFace.resize(faceRect.width, faceRect.height);
-      const faceRegion = img.getRegion(faceRect);
-      const replacementBuffer = resizedReplacementFace.bitmap.data;
+      const resizedReplacementFace = replacementFace.resize(faceRect.width, faceRect.height)
+      const faceRegion = img.getRegion(faceRect)
+  
+      const replacementBuffer = resizedReplacementFace.bitmap.data
+      const centerX = faceRect.width / 2
+      const centerY = faceRect.height / 2
+      const maxRadius = Math.min(centerX, centerY)
       for (let y = 0; y < faceRect.height; y++) {
         for (let x = 0; x < faceRect.width; x++) {
-          const idx = (y * faceRect.width + x) << 2;
-          const [r, g, b, a] = replacementBuffer.slice(idx, idx + 4);
-          if (a > 0) {
-            faceRegion.set(y, x, new cv.Vec3(b, g, r));
+          const distance = Math.sqrt((x - centerX) ** 2 + (y - centerY) ** 2)
+          if (distance <= maxRadius) {
+            const idx = (y * faceRect.width + x) << 2
+            const [r, g, b, a] = replacementBuffer.slice(idx, idx + 4)
+  
+            if (a > 0) {
+              faceRegion.set(y, x, new cv.Vec3(b, g, r))
+            }
           }
         }
       }
-    });
-
+    })
     const outputBuffer = cv.imencode('.jpg', img);
     return outputBuffer;
   } catch (error) {
@@ -90,20 +98,28 @@ async function processImage(imageBuffer, classifiersFolder, replacementImagePath
 
 // Express接口
 app.get('/process-image', async (req, res) => {
-  const { imageUrl } = req.query;
+  const { imageUrl, replacementImageUrl } = req.query;
 
-  if (!imageUrl) {
-    return res.status(400).send('Image URL is required');
+  if (!imageUrl || !replacementImageUrl) {
+    return res.status(400).send('Both image URL and replacement image URL are required');
   }
 
   const classifiersFolder = './classifiers';
-  const replacementImagePath = './replacement_face.png';
-
+  let replacementImageBuffer;
   try {
-    // 下载并转换输入图片到内存中
+    // 下载并转换输入图片和替换图片到内存中
     const imageBuffer = await downloadAndConvertImage(imageUrl);
+    if (replacementImageUrl === 'replace') {
+      // 使用默认的本地替换图像路径
+      const defaultReplacementImagePath = './replacement_face.png';
+      replacementImageBuffer = await Jimp.read(defaultReplacementImagePath);
+    } else {
+      // 下载并转换替换图片到内存中
+      replacementImageBuffer = await downloadAndConvertImage(replacementImageUrl);
+    }
+
     // 处理图片
-    const outputBuffer = await processImage(imageBuffer, classifiersFolder, replacementImagePath);
+    const outputBuffer = await processImage(imageBuffer, classifiersFolder, replacementImageBuffer);
 
     // 返回处理后的图片
     res.set('Content-Type', 'image/jpeg');
@@ -116,4 +132,4 @@ app.get('/process-image', async (req, res) => {
 
 app.listen(port, () => {
   console.log(`Server is running on http://localhost:${port}`);
-});
+});