luoyutianyang commited on
Commit
f4c6236
·
verified ·
1 Parent(s): ba3f4e4

Delete egg-api.js

Browse files
Files changed (1) hide show
  1. egg-api.js +0 -128
egg-api.js DELETED
@@ -1,128 +0,0 @@
1
- const express = require('express');
2
- const fs = require('fs');
3
- const path = require('path');
4
- const cv = require('opencv4nodejs');
5
- const Jimp = require('jimp');
6
- const axios = require('axios');
7
- const sharp = require('sharp');
8
-
9
- const app = express();
10
- const port = 7860; // 将端口改为 7860
11
- const outputDir = './output';
12
-
13
- // 确保输出目录存在
14
- if (!fs.existsSync(outputDir)) {
15
- fs.mkdirSync(outputDir);
16
- }
17
-
18
- // 并行处理图片下载和处理
19
- function processImage(url, inputImagePath, outputImagePath, classifiersFolder, replacementImagePath, callback) {
20
- axios.get(url, { responseType: 'arraybuffer' })
21
- .then(response => {
22
- return sharp(response.data).toFile(inputImagePath);
23
- })
24
- .then(() => {
25
- return replaceFaces(inputImagePath, outputImagePath, classifiersFolder, replacementImagePath);
26
- })
27
- .then(() => {
28
- console.log('Image processed successfully.');
29
- callback(null);
30
- })
31
- .catch(error => {
32
- console.error('Error processing image:', error);
33
- callback(error);
34
- });
35
- }
36
-
37
- // 加载所有分类器
38
- function loadClassifiersFromFolder(folderPath, callback) {
39
- const classifiers = [];
40
- fs.readdir(folderPath, (err, files) => {
41
- if (err) {
42
- callback(err);
43
- return;
44
- }
45
- let completed = 0;
46
- files.forEach(file => {
47
- const filePath = path.join(folderPath, file);
48
- const classifier = new cv.CascadeClassifier(filePath);
49
- classifiers.push(classifier);
50
- completed++;
51
- if (completed === files.length) {
52
- callback(null, classifiers);
53
- }
54
- });
55
- });
56
- }
57
-
58
- // 人脸替换函数
59
- function replaceFaces(inputImagePath, outputImagePath, classifiersFolder, replacementImagePath, callback) {
60
- cv.imreadAsync(inputImagePath, (err, img) => {
61
- if (err) {
62
- callback(err);
63
- return;
64
- }
65
- loadClassifiersFromFolder(classifiersFolder, (err, classifiers) => {
66
- if (err) {
67
- callback(err);
68
- return;
69
- }
70
- Jimp.read(replacementImagePath, (err, replacementFace) => {
71
- if (err) {
72
- callback(err);
73
- return;
74
- }
75
- const grayImg = img.bgrToGray();
76
- const allFaces = [];
77
- classifiers.forEach(classifier => {
78
- const faces = classifier.detectMultiScale(grayImg).objects;
79
- allFaces.push(...faces);
80
- });
81
-
82
- allFaces.forEach(faceRect => {
83
- const resizedReplacementFace = replacementFace.resize(faceRect.width, faceRect.height);
84
- const faceRegion = img.getRegion(faceRect);
85
- const replacementBuffer = resizedReplacementFace.bitmap.data;
86
- for (let y = 0; y < faceRect.height; y++) {
87
- for (let x = 0; x < faceRect.width; x++) {
88
- const idx = (y * faceRect.width + x) << 2;
89
- const [r, g, b, a] = replacementBuffer.slice(idx, idx + 4);
90
- if (a > 0) {
91
- faceRegion.set(y, x, new cv.Vec3(b, g, r));
92
- }
93
- }
94
- }
95
- });
96
-
97
- cv.imwrite(outputImagePath, img);
98
- callback(null);
99
- });
100
- });
101
- });
102
- }
103
-
104
- // Express接口
105
- app.get('/process-image', (req, res) => {
106
- const { imageUrl } = req.query;
107
-
108
- if (!imageUrl) {
109
- return res.status(400).send('Image URL is required');
110
- }
111
-
112
- const inputImagePath = path.join(outputDir, 'input.jpg'); // 使用jpg格式保存
113
- const outputImagePath = path.join(outputDir, 'output.jpg');
114
- const replacementImagePath = './replacement_face.png';
115
- const classifiersFolder = './classifiers';
116
-
117
- processImage(imageUrl, inputImagePath, outputImagePath, classifiersFolder, replacementImagePath, (err) => {
118
- if (err) {
119
- console.error('Error processing image:', err);
120
- return res.status(500).send('Error processing image');
121
- }
122
- res.sendFile(outputImagePath, { root: '.' });
123
- });
124
- });
125
-
126
- app.listen(port, () => {
127
- console.log(`Server is running on http://localhost:${port}`);
128
- });