docs: Update README.md to reflect changes in NSFWClassifierNode and WatermarkDetectionNode inputs and outputs, and add torchvision to installation instructions

This commit is contained in:
Craig Wright
2025-03-26 17:34:38 +00:00
parent 024fdf44b4
commit 9d64aad7cc
+14 -5
View File
@@ -46,13 +46,21 @@ ComfyUI-LexTools is a Python-based image processing and analysis toolkit that us
- _Input_: `image` (IMAGE)
- _Output_: Document type index and name.
- `NSFWClassifierNode`: Classifies content safety levels.
- _Input_: `image` (IMAGE), `show_on_node` (BOOL)
- _Input_: `image` (IMAGE), `show_on_node` (BOOL), `threshold` (FLOAT)
- _Output_:
- Classification report (STRING)
- SFW Score (FLOAT)
- NSFW Score (FLOAT)
- Neutral Score (FLOAT)
- Sexy Score (FLOAT)
- Porn Score (FLOAT)
- Is SFW (BOOLEAN)
- Is NSFW (BOOLEAN)
- `WatermarkDetectionNode`: Detects watermarks in images using EfficientNet.
- _Input_: `image` (IMAGE), `show_on_node` (BOOL), `threshold` (FLOAT)
- _Output_:
- Classification report (STRING)
- Clean Score (FLOAT)
- Watermark Score (FLOAT)
- Is Clean (BOOLEAN)
- Has Watermark (BOOLEAN)
3. **SegformerNode.py** - Handles semantic segmentation of images:
- `SegformerNode`: Performs semantic segmentation with multiple model options.
@@ -82,12 +90,13 @@ The project requires the following Python libraries:
- numpy
- scipy
- huggingface_hub
- torchvision
## Installation
1. Install the required Python packages:
```bash
pip install torch transformers pillow matplotlib numpy scipy huggingface_hub
pip install torch transformers pillow matplotlib numpy scipy huggingface_hub torchvision
```
2. Clone this repository into your ComfyUI custom_nodes directory: