Compare commits

...

48 Commits

Author SHA1 Message Date
f7c4acfebb Remove FastAbsoluteSaver node (moved to ComfyUI-JSON-Dynamic)
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-02-26 16:11:38 +01:00
6c7f618bb0 Only add sharpness score metadata when scores are actually connected
Return None for scores_list when scores_info input is not connected,
and skip writing score metadata in that case for both images and videos.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-02-05 11:26:36 +01:00
f02851d88a Fix argument too long error by using ffmetadata file for video metadata
Write metadata to a temp file and pass via -i/-map_metadata instead of
command-line -metadata flags to avoid OS argument length limits with
large ComfyUI workflow JSON.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-02-03 00:11:20 +01:00
59bd805920 Fix ValueError flush of closed file in video encoding
Replace communicate() with direct stderr.read() + wait() to avoid
double-closing stdin. Catch BrokenPipeError for early ffmpeg exits.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-02-02 23:12:44 +01:00
3297f3a203 Add metadata embedding to video output
Writes sharpness scores (avg + per-frame) and optionally the ComfyUI
workflow/prompt as ffmpeg container metadata in mp4/webm files.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-02-02 23:01:56 +01:00
3b87ac820f Add video saving support (mp4/webm) to FastAbsoluteSaver
Adds mp4 (H.264) and webm (VP9) output formats with configurable FPS,
CRF, and pixel format. Includes auto-discovery of ffmpeg from bundled
binary, imageio_ffmpeg, system PATH, or automatic static download.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-02-02 23:00:17 +01:00
b23773c7c2 Update fast_saver.py 2026-01-21 11:21:00 +01:00
8560f24d36 Update fast_saver.py 2026-01-21 10:45:09 +01:00
c40c1fd82c Update fast_saver.py 2026-01-20 01:00:03 +01:00
07acefffc1 Update README.md 2026-01-20 00:46:59 +01:00
b30e2d0233 Update README.md 2026-01-20 00:40:53 +01:00
dd6a9aefd7 Upload files to "assets" 2026-01-20 00:39:52 +01:00
60362e3514 Upload files to "example_workflows" 2026-01-20 00:39:26 +01:00
f03979a767 Delete example_workflows/comfyui-sharp-example.json 2026-01-20 00:38:06 +01:00
c32a4bcb32 Update README.md 2026-01-20 00:34:47 +01:00
162699a4a2 Merge pull request 'webp' (#4) from webp into main
Reviewed-on: #4
2026-01-20 00:31:09 +01:00
f63b837a2c Update fast_saver.py 2026-01-20 00:21:42 +01:00
b54b4329ca Update fast_saver.py 2026-01-20 00:20:21 +01:00
1bde14bd97 Update fast_saver.py 2026-01-20 00:08:41 +01:00
a2d79a7e6c Merge pull request 'fast-saver' (#3) from fast-saver into main
Reviewed-on: #3
2026-01-19 23:57:45 +01:00
24a59a6da2 Update fast_saver.py 2026-01-19 23:44:40 +01:00
099ce948ae Update __init__.py 2026-01-19 23:09:21 +01:00
44f3130a15 Update __init__.py 2026-01-19 23:05:03 +01:00
178247c79f Add fast_saver.py 2026-01-19 23:04:07 +01:00
3d53b94435 Update README.md 2026-01-19 17:28:56 +01:00
e762c0b90f Update readme 2026-01-19 17:27:18 +01:00
6d5c773bea Update parallel_loader.py 2026-01-19 13:24:52 +01:00
6eccb258ee Merge pull request 'parralel' (#2) from parralel into main
Reviewed-on: #2
2026-01-19 12:30:42 +01:00
dd95977b2a Update readme 2026-01-19 12:30:17 +01:00
dc77f49d8e Update readme 2026-01-19 12:29:04 +01:00
bf7da8f4b3 Update readme 2026-01-19 12:28:45 +01:00
1c1f566e5f Update readme 2026-01-19 12:26:39 +01:00
56e6df35e8 Delete example_workflows/nodes.png 2026-01-19 12:26:16 +01:00
6dbf340171 Upload files to "assets" 2026-01-19 12:26:04 +01:00
7639c3c158 Update readme 2026-01-19 12:23:56 +01:00
7688674563 Upload files to "example_workflows" 2026-01-19 12:22:23 +01:00
2a82ef8201 Update readme 2026-01-19 12:17:50 +01:00
033fc4a626 Update readme 2026-01-19 12:17:21 +01:00
ef92a569a0 Delete example_workflows/comfyui-sharp.json 2026-01-19 12:14:27 +01:00
008f8edfad Upload files to "example_workflows" 2026-01-19 12:14:18 +01:00
610c52c0b8 Update parallel_loader.py 2026-01-19 12:00:54 +01:00
689c67cd7f Update parallel_loader.py 2026-01-19 11:54:46 +01:00
50f2080c8b Update __init__.py 2026-01-19 11:42:47 +01:00
2cc185c000 Update parallel_loader.py 2026-01-19 11:39:59 +01:00
56629c5490 Update __init__.py 2026-01-19 11:34:37 +01:00
dab38a1fbf Delete js/sharp_tooltips.js 2026-01-19 11:19:49 +01:00
c0952ecbf1 Update __init__.py 2026-01-19 11:19:30 +01:00
ca7ca06791 Add parallel_loader.py 2026-01-19 11:17:28 +01:00
9 changed files with 654 additions and 765 deletions

1
.gitignore vendored Normal file
View File

@@ -0,0 +1 @@
ffmpeg_bin/

142
README.md Normal file
View File

@@ -0,0 +1,142 @@
# ComfyUI Sharpness Helper Nodes
A high-performance custom node suite for **ComfyUI** designed to detect blur, calculate sharpness scores (Laplacian Variance), and efficiently extract or filter the best frames from videos and image batches.
This pack is for a personnal project:
1. **Dataset Creation:** Extracting only the sharpest frames from massive movie files without crashing RAM.
2. **Generation Filtering:** Automatically discarding blurry frames from Wan or img2img outputs.
---
![node](assets/nodes.png)
---
## 🚀 Key Features
### 1. Parallel Video Loader (Path-Based)
* **Zero-RAM Scanning:** Scans video files directly from disk without decoding every frame to memory.
* **Multi-Threaded:** Uses all CPU cores to calculate sharpness scores at high speed (1000s of frames per minute).
* **Smart Batching:** Includes an auto-incrementing "Page" system to process long movies in chunks (e.g., minute-by-minute) without restarting ComfyUI.
* **Lazy Loading:** Only decodes and loads the final "Best N" frames into ComfyUI tensors.
### 2. Fast Absolute Saver (Metadata)
* **Multi-Threaded Saving:** Spawns parallel workers to saturate SSD write speeds (bypassing standard PIL bottlenecks).
* **No UI Lag:** Saves images in the background without trying to render Base64 previews in the browser, preventing interface freezes.
* **Metadata Embedding:** Automatically embeds the sharpness score into the PNG/WebP metadata for dataset curation.
* **Smart Naming:** Uses original video frame numbers in filenames (e.g., `frame_001450.png`) instead of arbitrary counters.
### 3. Standard Sharpness Duo (Tensor-Based)
* **Workflow Integration:** Works with any node that outputs an `IMAGE` batch (e.g., AnimateDiff, VideoHelperSuite).
* **Precision Filtering:** Sorts and filters generated frames before saving or passing to a second pass (img2img).
---
## 📦 Installation
1. Clone this repository into your `custom_nodes` folder:
```bash
cd ComfyUI/custom_nodes/
git clone [https://github.com/YOUR_USERNAME/ComfyUI-Sharpness-Helper.git](https://github.com/YOUR_USERNAME/ComfyUI-Sharpness-Helper.git)
```
2. Install dependencies (if needed):
```bash
pip install opencv-python numpy
```
3. Restart ComfyUI.
---
## 🛠️ Node Documentation
### 1. Parallel Video Loader (Sharpness)
**Category:** `BetaHelper/Video`
This is the recommended node for **Dataset Creation** or finding good frames in **Long Movies**. It inputs a file path, scans it in parallel, and only loads the final "Best N" frames into memory.
| Input | Description |
| :--- | :--- |
| **video_path** | Absolute path to your video file (e.g., `D:\Movies\input.mp4`). |
| **batch_index** | **Critical.** Connect a **Primitive Node** here set to `increment`. This controls which "chunk" of the video you are viewing. |
| **scan_limit** | How many frames to process per batch (e.g., `1440`). |
| **frame_scan_step** | Speed up scanning by checking every Nth frame (e.g., `5` checks frames 0, 5, 10...). |
| **manual_skip_start** | Global offset (e.g., set to `2000` to always ignore the opening credits). |
**Outputs:**
* `images`: The batch of the sharpest frames found.
* `scores_info`: String containing frame indices and scores (Connect to Saver).
* `batch_int`: The current batch number.
* `batch_status`: Human-readable status (e.g., *"Batch 2: Skipped 2880 frames..."*).
> **💡 Pro Tip:** To scan a movie continuously, connect a **Primitive Node** to `batch_index`, set it to **increment**, and enable "Auto Queue" in ComfyUI.
---
### 2. Fast Absolute Saver (Metadata)
**Category:** `BetaHelper/IO`
A "Pro-Grade" saver designed for speed. It bypasses relative paths and UI previews.
| Input | Description |
| :--- | :--- |
| **output_path** | Absolute path to save folder (e.g., `D:\Datasets\Sharp_Output`). |
| **filename_prefix** | Base name for files (e.g., `matrix_movie`). |
| **max_threads** | **0 = Auto** (Uses all CPU cores). Set manually to limit CPU usage. |
| **save_format** | `png` (Fastest) or `webp` (Smaller size). |
| **filename_with_score** | If True, appends score to filename: `frame_001450_1500.png`. |
| **scores_info** | Connect this to the `scores_info` output of the Parallel Loader to enable smart naming. |
**Performance Note:**
* **PNG:** Uses `compress_level=1` for maximum speed.
* **WebP:** Avoid `webp_method=6` unless you need max compression; it is very CPU intensive. `4` is the recommended balance.
---
### 3. Sharpness Analyzer & Selector (The Duo)
**Category:** `BetaHelper/Image`
Use these when you already have images inside your workflow (e.g., from a generation or a standard Load Video node).
#### Node A: Sharpness Analyzer
* **Input:** `IMAGE` batch.
* **Action:** Calculates the Laplacian Variance for every image in the batch.
* **Output:** Passes the images through + a generic score list.
#### Node B: SharpFrame Selector
* **Input:** `IMAGE` batch (from Analyzer).
* **Action:** Sorts the batch based on the scores and picks the top N frames.
* **Output:** A reduced batch containing only the sharpest images.
---
## ⚖️ Which Node Should I Use?
| Feature | **Parallel Video Loader** | **Standard Duo** |
| :--- | :--- | :--- |
| **Input Type** | File Path (`String`) | Image Tensor (`IMAGE`) |
| **Best For** | **Long Videos / Movies** | **Generations / Short Clips** |
| **Memory Usage** | Very Low (Only loads final frames) | High (Loads all frames to RAM first) |
| **Speed** | ⚡ **Ultra Fast** (Multi-core) | 🐢 Standard (Single-core) |
| **Workflow Stage** | Start of Workflow | Middle/End of Workflow |
---
## 📝 Example Workflows
### Batch Processing a Movie for Training Data
1. Add **Parallel Video Loader**.
2. Connect a **Primitive Node** to `batch_index` (Control: `increment`).
3. Set `scan_limit` to `1000` and `frame_scan_step` to `5`.
4. Connect `images` and `scores_info` to **Fast Absolute Saver**.
5. Enable **Auto Queue** in ComfyUI extra options.
* *Result: ComfyUI will loop through your movie, extracting the 4 sharpest frames from every ~1000 frame chunk automatically.*
### Filtering AnimateDiff Output
1. AnimateDiff Generation -> **Sharpness Analyzer**.
2. Analyzer Output -> **SharpFrame Selector** (Select Best 1).
3. Selector Output -> **Face Detailer** or **Upscaler**.
* *Result: Only the clearest frame from your animation is sent to the upscaler, saving time on blurry frames.*
---
## Credits
* Built using `opencv-python` for Laplacian Variance calculation.
* Parallel processing logic for efficient large-file handling.

View File

@@ -1,13 +1,16 @@
from .sharp_node import SharpnessAnalyzer, SharpFrameSelector from .sharp_node import SharpnessAnalyzer, SharpFrameSelector
from .parallel_loader import ParallelSharpnessLoader
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"SharpnessAnalyzer": SharpnessAnalyzer, "SharpnessAnalyzer": SharpnessAnalyzer,
"SharpFrameSelector": SharpFrameSelector "SharpFrameSelector": SharpFrameSelector,
"ParallelSharpnessLoader": ParallelSharpnessLoader,
} }
NODE_DISPLAY_NAME_MAPPINGS = { NODE_DISPLAY_NAME_MAPPINGS = {
"SharpnessAnalyzer": "1. Sharpness Analyzer", "SharpnessAnalyzer": "1. Sharpness Analyzer",
"SharpFrameSelector": "2. Sharp Frame Selector" "SharpFrameSelector": "2. Sharp Frame Selector",
"ParallelSharpnessLoader": "3. Parallel Video Loader (Sharpness)",
} }
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]

BIN
assets/nodes.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 347 KiB

View File

@@ -0,0 +1,378 @@
{
"id": "4fbf6f31-0f7b-4465-8ec8-25df4862e076",
"revision": 0,
"last_node_id": 35,
"last_link_id": 44,
"nodes": [
{
"id": 31,
"type": "PrimitiveNode",
"pos": [
4672,
-928
],
"size": [
210,
82
],
"flags": {},
"order": 0,
"mode": 0,
"inputs": [],
"outputs": [
{
"name": "connect to widget input",
"type": "*",
"links": []
}
],
"properties": {
"Run widget replace on values": false,
"ue_properties": {
"widget_ue_connectable": {},
"version": "7.5.2",
"input_ue_unconnectable": {}
}
}
},
{
"id": 1,
"type": "SharpFrameSelector",
"pos": [
4992,
-704
],
"size": [
288,
174
],
"flags": {},
"order": 4,
"mode": 0,
"inputs": [
{
"name": "images",
"type": "IMAGE",
"link": null
},
{
"name": "scores",
"type": "SHARPNESS_SCORES",
"link": 3
}
],
"outputs": [
{
"name": "selected_images",
"type": "IMAGE",
"links": [
32
]
},
{
"name": "count",
"type": "INT",
"links": null
}
],
"properties": {
"aux_id": "ComfyUI-Sharp-Selector.git",
"ver": "f30f948c9fa8acf9b7fe09559f172d8a63468c8d",
"Node name for S&R": "SharpFrameSelector",
"ue_properties": {
"widget_ue_connectable": {},
"input_ue_unconnectable": {},
"version": "7.5.2"
}
},
"widgets_values": [
"best_n",
144,
24,
3,
0
]
},
{
"id": 32,
"type": "easy showAnything",
"pos": [
5344,
-1024
],
"size": [
448,
96
],
"flags": {},
"order": 6,
"mode": 0,
"inputs": [
{
"name": "anything",
"shape": 7,
"type": "*",
"link": 39
}
],
"outputs": [
{
"name": "output",
"type": "*",
"links": null
}
],
"properties": {
"cnr_id": "comfyui-easy-use",
"ver": "5dfcbcf51d8a6efed947bc7bdd6797827fecab55",
"Node name for S&R": "easy showAnything",
"ue_properties": {
"widget_ue_connectable": {},
"input_ue_unconnectable": {},
"version": "7.5.2"
}
},
"widgets_values": [
"Batch 1: Skipped 3440 frames. Scanning range 3440 -> 4880."
]
},
{
"id": 5,
"type": "SharpnessAnalyzer",
"pos": [
4672,
-672
],
"size": [
185.9771484375,
26
],
"flags": {},
"order": 1,
"mode": 0,
"inputs": [
{
"name": "images",
"type": "IMAGE",
"link": null
}
],
"outputs": [
{
"name": "scores",
"type": "SHARPNESS_SCORES",
"links": [
3,
21
]
}
],
"properties": {
"aux_id": "ComfyUI-Sharp-Selector.git",
"ver": "0df11447abb7f41bf7f12a2906aa868a5d2027b4",
"Node name for S&R": "SharpnessAnalyzer",
"ue_properties": {
"widget_ue_connectable": {},
"input_ue_unconnectable": {},
"version": "7.5.2"
}
},
"widgets_values": []
},
{
"id": 35,
"type": "FastAbsoluteSaver",
"pos": [
5856,
-1024
],
"size": [
306.3776153564453,
270
],
"flags": {},
"order": 5,
"mode": 0,
"inputs": [
{
"name": "images",
"type": "IMAGE",
"link": 43
},
{
"name": "scores_info",
"shape": 7,
"type": "STRING",
"link": 44
}
],
"outputs": [],
"properties": {
"aux_id": "ComfyUI-Sharp-Selector.git",
"ver": "162699a4a23219ac5ac75f398a17e67c3767da46",
"ue_properties": {
"widget_ue_connectable": {},
"input_ue_unconnectable": {}
},
"Node name for S&R": "FastAbsoluteSaver"
},
"widgets_values": [
"D:\\Datasets\\Sharp_Output",
"frame",
"png",
0,
false,
"sharpness_score",
true,
100,
4
]
},
{
"id": 29,
"type": "ParallelSharpnessLoader",
"pos": [
4992,
-1024
],
"size": [
320,
262
],
"flags": {},
"order": 2,
"mode": 0,
"inputs": [],
"outputs": [
{
"name": "images",
"type": "IMAGE",
"links": [
43
]
},
{
"name": "scores_info",
"type": "STRING",
"links": [
44
]
},
{
"name": "batch_int",
"type": "INT",
"links": null
},
{
"name": "batch_status",
"type": "STRING",
"links": [
39
]
}
],
"properties": {
"aux_id": "ComfyUI-Sharp-Selector.git",
"ver": "dab38a1fbf0077655fe568d500866fce6ecc857d",
"Node name for S&R": "ParallelSharpnessLoader",
"ue_properties": {
"widget_ue_connectable": {},
"input_ue_unconnectable": {},
"version": "7.5.2"
}
},
"widgets_values": [
"C:\\path\\to\\video.mp4",
0,
1440,
1,
30,
24,
2000
]
},
{
"id": 34,
"type": "Note",
"pos": [
4224,
-1120
],
"size": [
416,
736
],
"flags": {},
"order": 3,
"mode": 0,
"inputs": [],
"outputs": [],
"properties": {
"ue_properties": {
"widget_ue_connectable": {},
"version": "7.5.2",
"input_ue_unconnectable": {}
}
},
"widgets_values": [
"📝 Smart Dataset Extraction Workflow\n\n1. Parallel Video Loader (The Source)\n\n What it does: Scans your video file directly from the hard drive using multi-threading. It does not load the whole video into RAM.\n\n Batching: Uses the batch_index (Primitive Node) to \"page\" through the movie.\n\n Example: If scan_limit is 1440, Batch 0 scans frames 0-1440, Batch 1 scans 1440-2880, etc.\n\n Selection: It calculates sharpness (Laplacian Variance) and only decodes the \"Best N\" frames to send downstream.\n\n2. Fast Absolute Saver (The Destination)\n\n What it does: Saves images instantly to your SSD using parallel workers, bypassing the slow ComfyUI preview window.\n\n Smart Naming: Connect scores_info from the Loader to the Saver! This allows files to be named using the original video frame number (e.g., movie_frame_00450.png) rather than a random batch counter.\n\n Metadata: Embeds the sharpness score into the PNG/WebP metadata for future filtering.\n\n⚠ Usage Tips:\n\n Automation: Set batch_index to \"Increment\" (on the Primitive Node) and enable \"Auto Queue\" in ComfyUI options to process the entire movie automatically.\n\n Monitoring: Watch the Console Window (black command prompt) for progress logs. The saver does not preview images in the UI to prevent lag.\n\n Safety: The saver uses absolute paths and overwrites files with the same name. Use a unique filename_prefix for each new video source."
],
"color": "#432",
"bgcolor": "#653"
}
],
"links": [
[
3,
5,
0,
1,
1,
"SHARPNESS_SCORES"
],
[
39,
29,
3,
32,
0,
"STRING"
],
[
43,
29,
0,
35,
0,
"IMAGE"
],
[
44,
29,
1,
35,
1,
"STRING"
]
],
"groups": [],
"config": {},
"extra": {
"workflowRendererVersion": "LG",
"ue_links": [],
"links_added_by_ue": [],
"ds": {
"scale": 1.1,
"offset": [
-3048.698587382934,
1363.985488079904
]
},
"frontendVersion": "1.36.14",
"VHS_latentpreview": true,
"VHS_latentpreviewrate": 0,
"VHS_MetadataImage": true,
"VHS_KeepIntermediate": true
},
"version": 0.4
}

File diff suppressed because one or more lines are too long

View File

@@ -1,35 +0,0 @@
import { app } from "../../scripts/app.js";
app.registerExtension({
name: "SharpFrames.Tooltips",
async beforeRegisterNodeDef(nodeType, nodeData, app) {
if (nodeData.name === "SharpFrameSelector") {
const tooltips = {
// Must match Python INPUT_TYPES keys exactly
"selection_method": "Strategy:\n• 'batched': Best for video. Splits time into slots.\n• 'best_n': Global top sharpest frames.",
"batch_size": "For 'batched' mode.\nSize of the analysis window (in frames).",
"batch_buffer": "For 'batched' mode.\nFrames to skip AFTER each batch (dead zone).",
"num_frames": "For 'best_n' mode.\nTotal frames to output.",
"min_sharpness": "Threshold Filter.\nDiscard frames with score below this.\nNote: Scores are lower on resized images.",
"images": "Input High-Res images.",
"scores": "Input Sharpness Scores from Analyzer."
};
const onNodeCreated = nodeType.prototype.onNodeCreated;
nodeType.prototype.onNodeCreated = function () {
onNodeCreated?.apply(this, arguments);
if (this.widgets) {
for (const w of this.widgets) {
if (tooltips[w.name]) {
w.tooltip = tooltips[w.name];
// Force update for immediate feedback
w.options = w.options || {};
w.options.tooltip = tooltips[w.name];
}
}
}
};
}
},
});

128
parallel_loader.py Normal file
View File

@@ -0,0 +1,128 @@
import cv2
import torch
import numpy as np
import concurrent.futures
import os
class ParallelSharpnessLoader:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"video_path": ("STRING", {"default": "C:\\path\\to\\video.mp4"}),
# BATCHING CONTROLS
"batch_index": ("INT", {"default": 0, "min": 0, "max": 10000, "step": 1, "label": "Batch Counter (Auto-Increment)"}),
"scan_limit": ("INT", {"default": 1440, "min": 1, "max": 10000000, "step": 1, "label": "Frames per Batch"}),
# STANDARD CONTROLS
"frame_scan_step": ("INT", {"default": 5, "min": 1, "step": 1, "label": "Analyze Every Nth Frame"}),
"return_count": ("INT", {"default": 4, "min": 1, "max": 1024, "step": 1, "label": "Best Frames to Return"}),
"min_distance": ("INT", {"default": 24, "min": 0, "max": 10000, "step": 1, "label": "Min Distance (Frames)"}),
"manual_skip_start": ("INT", {"default": 0, "min": 0, "max": 10000000, "step": 1, "label": "Global Start Offset"}),
},
}
RETURN_TYPES = ("IMAGE", "STRING", "INT", "STRING")
RETURN_NAMES = ("images", "scores_info", "batch_int", "batch_status")
FUNCTION = "load_video"
CATEGORY = "BetaHelper/Video"
def calculate_sharpness(self, frame_data):
gray = cv2.cvtColor(frame_data, cv2.COLOR_BGR2GRAY)
return cv2.Laplacian(gray, cv2.CV_64F).var()
def load_video(self, video_path, batch_index, scan_limit, frame_scan_step, return_count, min_distance, manual_skip_start):
# 1. Validation
if not os.path.exists(video_path):
video_path = video_path.strip('"')
if not os.path.exists(video_path):
raise FileNotFoundError(f"Video not found: {video_path}")
# 2. Calculate Offsets
current_skip = (batch_index * scan_limit) + manual_skip_start
range_end = current_skip + scan_limit
status_msg = f"Batch {batch_index}: Skipped {current_skip} frames. Scanning range {current_skip} -> {range_end}."
print(f"xx- Parallel Loader | {status_msg}")
cap = cv2.VideoCapture(video_path)
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
# --- STOP CONDITION 1: REACHED END OF VIDEO ---
# This stops the queue immediately if we try to read past the end.
if current_skip >= total_frames:
cap.release()
raise ValueError(f"Processing Complete. Batch {batch_index} starts at frame {current_skip}, but video only has {total_frames} frames.")
# 3. Scanning (Pass 1)
if current_skip > 0:
cap.set(cv2.CAP_PROP_POS_FRAMES, current_skip)
frame_scores = []
current_frame = current_skip
scanned_count = 0
with concurrent.futures.ThreadPoolExecutor(max_workers=16) as executor:
futures = []
while True:
if scanned_count >= scan_limit:
break
ret, frame = cap.read()
if not ret: break
future = executor.submit(self.calculate_sharpness, frame)
futures.append((current_frame, future))
scanned_count += 1
# Manual Stepping
if frame_scan_step > 1:
for _ in range(frame_scan_step - 1):
if not cap.grab(): break
current_frame += 1
current_frame += 1
for idx, future in futures:
frame_scores.append((idx, future.result()))
cap.release()
# 4. Selection
# --- STOP CONDITION 2: NO FRAMES FOUND ---
if not frame_scores:
raise ValueError(f"No frames found in batch {batch_index} (Range {current_skip}-{range_end}). The video might be corrupted or blank.")
frame_scores.sort(key=lambda x: x[1], reverse=True)
selected = []
for idx, score in frame_scores:
if len(selected) >= return_count: break
if all(abs(s[0] - idx) >= min_distance for s in selected):
selected.append((idx, score))
selected.sort(key=lambda x: x[0])
# 5. Extraction (Pass 2)
cap = cv2.VideoCapture(video_path)
output_tensors = []
info_log = []
for idx, score in selected:
cap.set(cv2.CAP_PROP_POS_FRAMES, idx)
ret, frame = cap.read()
if ret:
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
frame = frame.astype(np.float32) / 255.0
output_tensors.append(torch.from_numpy(frame))
info_log.append(f"F:{idx} (Score:{int(score)})")
cap.release()
if not output_tensors:
raise ValueError("Frames were selected but could not be loaded. This indicates a file read error.")
return (torch.stack(output_tensors), ", ".join(info_log), batch_index, status_msg)

29
readme
View File

@@ -1,29 +0,0 @@
# 🔪 ComfyUI Sharp Frame Selector
A suite of custom nodes for [ComfyUI](https://github.com/comfyanonymous/ComfyUI) designed to intelligently extract the sharpest frames from video footage.
Based on the [sharp-frames](https://github.com/Reflct/sharp-frames-python) logic, this tool uses **Laplacian Variance** to score image clarity. It is optimized for high-resolution video processing using a **Sidechain Workflow** that saves massive amounts of RAM.
## ✨ Key Features
* **Sidechain Optimization:** Analyze lightweight 512px proxy images to control the selection of heavy 4K raw frames.
* **Batched Extraction:** Splits video into time slots (e.g., 1 second) and picks the single best frame from each slot. Perfect for ensuring action scenes are not missed.
* **Threshold Filtering:** Automatically discards frames that are too blurry, even if they are the "winner" of their batch.
* **Buffer Control:** Optional dead-zones between batches to reduce frame count or ensure temporal separation.
---
## 🚀 Installation
### Option 1: ComfyUI Manager (Recommended)
1. Open ComfyUI Manager.
2. Search for **"Sharp Frame Selector"**.
3. Click **Install**.
### Option 2: Manual Installation
Clone this repository into your `custom_nodes` folder:
```bash
cd ComfyUI/custom_nodes/
git clone [https://github.com/YOUR_USERNAME/ComfyUI-Sharp-Selector.git](https://github.com/YOUR_USERNAME/ComfyUI-Sharp-Selector.git)
pip install -r ComfyUI-Sharp-Selector/requirements.txt