xueteng/video-flow-b
1
0
Fork 0
forked from 77media/video-flow
Code Pull Requests Activity

修复问题，人脸识别数据

Browse Source
This commit is contained in:
海龙 2025-08-14 17:06:26 +08:00
parent 1c57d04ed1
commit 09137b23e3
3 changed files with 8 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
api/DTO/movieEdit.ts
View File

@ -541,19 +541,7 @@ export interface MatchedPerson {
/**高度 小数百分比形式 */ /**高度 小数百分比形式 */
height: number; height: number;
} }
/**
* 角色识别结果数据
* @description AI识别图片中人物后的详细结果
*/
export interface RecognitionResultData {
/** 目标图片URL */
target_image_url: string;
/** 检测到的总人数 */
total_persons_detected: number;
/** 匹配的人物列表 */
matched_persons: MatchedPerson[];
}
/** /**
* 角色识别响应结果 * 角色识别响应结果
* @description API返回的角色识别完整响应 * @description API返回的角色识别完整响应
@ -565,7 +553,7 @@ export interface RecognitionResult {
/** 响应消息 */ /** 响应消息 */
message: string; message: string;
/** 识别结果数据 */ /** 识别结果数据 */
data: RecognitionResultData; data: MatchedPerson[];
} }
/** /**
* 使用的角色信息 * 使用的角色信息

16
app/service/Interaction/ShotService.ts
View File

@ -402,7 +402,7 @@ export const useShotService = (): UseShotService => {
); );
console.log("角色识别结果:", recognitionResult); console.log("角色识别结果:", recognitionResult);
setMatched_persons( setMatched_persons(
recognitionResult.recognition_result.data.matched_persons recognitionResult.recognition_result.data
); );
return recognitionResult; return recognitionResult;
} catch (recognitionError) { } catch (recognitionError) {
@ -441,21 +441,17 @@ export const useShotService = (): UseShotService => {
const containerRect = containerElement.getBoundingClientRect(); const containerRect = containerElement.getBoundingClientRect();
const containerWidth = containerRect.width; const containerWidth = containerRect.width;
const containerHeight = containerRect.height; const containerHeight = containerRect.height;
console.log('recognitionBoxes-width-height', containerWidth, containerHeight);
// 计算识别框属性 // 计算识别框属性
return matched_persons return matched_persons
.map((person) => { .map((person) => {
// 取出bbox信息
const bbox = (person as any).bbox;
if (!bbox) return null;
// 计算绝对坐标和尺寸（百分比转像素） // 计算绝对坐标和尺寸（百分比转像素）
const left = (bbox.x || 0) * containerWidth; const left = Number((person.x || 0).toFixed(4)) * containerWidth;
const top = (bbox.y || 0) * containerHeight; const top = Number((person.y || 0).toFixed(4)) * containerHeight;
const width = (bbox.width || 0) * containerWidth; const width = Number((person.width || 0).toFixed(4)) * containerWidth;
const height = (bbox.height || 0) * containerHeight; const height = Number((person.height || 0).toFixed(4)) * containerHeight;
console.log('left-top-width-height', left, top, width, height);
return { return {
left, left,
top, top,

2
components/ui/shot-tab-content.tsx
View File

@ -77,7 +77,7 @@ export function ShotTabContent({
const roleRecognitionResponse = await filterRole(containerElement); const roleRecognitionResponse = await filterRole(containerElement);
console.log('roleRecognitionResponse', roleRecognitionResponse); console.log('roleRecognitionResponse', roleRecognitionResponse);
if (roleRecognitionResponse && roleRecognitionResponse.recognition_result.code === 200) { if (roleRecognitionResponse && roleRecognitionResponse.recognition_result.code === 200) {
const recognitionBoxes = calculateRecognitionBoxes(containerElement, roleRecognitionResponse.recognition_result.data.matched_persons); const recognitionBoxes = calculateRecognitionBoxes(containerElement, roleRecognitionResponse.recognition_result.data);
console.log('recognitionBoxes', recognitionBoxes); console.log('recognitionBoxes', recognitionBoxes);
setDetections(recognitionBoxes.map((person: any) => ({ setDetections(recognitionBoxes.map((person: any) => ({
id: person.person_id, id: person.person_id,