Cherokee
968b9e6b19
- YOLOv8 person detection (ultralytics) - ByteTrack multi-object tracking (with greedy fallback) - InsightFace face recognition + cosine similarity clustering - Green screen actor cropping - Multi-segment video merging with ffmpeg - ComfyUI node: VideoActorExtractor (category: video/actor) - Output: per-actor MP4 + actor_info.json
39 lines
1.7 KiB
JavaScript
39 lines
1.7 KiB
JavaScript
// JavaScript widget for VideoActorExtract node
|
|
// Displays actor info JSON in a readable format
|
|
|
|
import { app } from "../../scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "VideoActorExtract.widget",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === "VideoActorExtractor" || nodeData.name === "Video Actor Extract") {
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function() {
|
|
if (onNodeCreated) {
|
|
onNodeCreated.apply(this, arguments);
|
|
}
|
|
this.addWidget("button", "View Output", "View", () => {
|
|
const jsonOutput = this.outputs?.[0]?.widget?.value;
|
|
if (jsonOutput) {
|
|
try {
|
|
const data = JSON.parse(jsonOutput);
|
|
let summary = "Actors found:\n";
|
|
for (const actor of data.actors || []) {
|
|
summary += ` ${actor.actor_id}: ${actor.segment_count} segment(s), ${actor.total_frames} frames\n`;
|
|
for (const seg of actor.segments || []) {
|
|
summary += ` [${seg.start_time_sec}s - ${seg.end_time_sec}s] (${seg.frame_count} frames)\n`;
|
|
}
|
|
}
|
|
alert(summary);
|
|
} catch (e) {
|
|
alert("Invalid JSON output: " + e.message);
|
|
}
|
|
} else {
|
|
alert("No output yet. Run the workflow first.");
|
|
}
|
|
});
|
|
};
|
|
}
|
|
},
|
|
});
|