Commit ·
fdaa122
1
Parent(s): b5e53f5
Updated README with metadata, tags, and test link
Browse filesCo-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
README.md
CHANGED
|
@@ -1,7 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# Fast Watermark Removal
|
| 2 |
|
| 3 |
A high-performance TorchScript model for removing watermarks from images. This model uses a dual-stage architecture optimized for speed and quality.
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
## Features
|
| 6 |
|
| 7 |
- **Fast inference**: ~500ms per image (RTX 4090)
|
|
@@ -9,6 +33,15 @@ A high-performance TorchScript model for removing watermarks from images. This m
|
|
| 9 |
- **Production-ready**: Compiled TorchScript model, no training code needed
|
| 10 |
- **Memory efficient**: Requires 11.5GB VRAM
|
| 11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
## Limitations
|
| 13 |
|
| 14 |
- **Output resolution**: Limited to 768px maximum dimension (aspect ratio preserved)
|
|
@@ -34,10 +67,6 @@ A commercial license with **1536px maximum output resolution** is available for
|
|
| 34 |
### Setup
|
| 35 |
|
| 36 |
```bash
|
| 37 |
-
# Clone the repository
|
| 38 |
-
git clone https://huggingface.co/[your-username]/remove-watermarks-fast
|
| 39 |
-
cd remove-watermarks-fast
|
| 40 |
-
|
| 41 |
# Install dependencies
|
| 42 |
pip install -r requirements.txt
|
| 43 |
```
|
|
@@ -53,7 +82,7 @@ python inference.py -i /path/to/watermarked/image.jpg -m model.ts -o output_fold
|
|
| 53 |
### Batch Processing
|
| 54 |
|
| 55 |
```bash
|
| 56 |
-
python inference.py -f /path/to/images/folder -m model.ts
|
| 57 |
```
|
| 58 |
|
| 59 |
### Arguments
|
|
@@ -82,24 +111,10 @@ The model uses a two-stage pipeline:
|
|
| 82 |
|
| 83 |
All processing (including resizing and normalization) is performed within the compiled TorchScript model for optimal performance.
|
| 84 |
|
| 85 |
-
## Performance
|
| 86 |
-
|
| 87 |
-
- **GPU**: NVIDIA RTX 3090 / A6000 or equivalent
|
| 88 |
-
- **VRAM**: 11.5GB required
|
| 89 |
-
- **Speed**: ~500ms per image (768px output)
|
| 90 |
-
- **Batch size**: 1 (optimized for low latency)
|
| 91 |
-
|
| 92 |
## Future Improvements
|
| 93 |
|
| 94 |
I'm actively exploring ways to enhance this model's capabilities. If you have suggestions, encounter issues, or are interested in collaborating on improvements, please reach out!
|
| 95 |
|
| 96 |
-
## Technical Details
|
| 97 |
-
|
| 98 |
-
- **Architecture**: Dual-stage with Swin2 Transformers
|
| 99 |
-
- **Format**: TorchScript (.ts) compiled model
|
| 100 |
-
- **Input**: RGB images (any resolution)
|
| 101 |
-
- **Output**: RGB images (max 768px, aspect ratio preserved)
|
| 102 |
-
- **Precision**: FP32 with TensorFloat32 matmul on Ampere+ GPUs
|
| 103 |
|
| 104 |
## License
|
| 105 |
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: other
|
| 3 |
+
license_name: non-commercial
|
| 4 |
+
license_link: LICENSE
|
| 5 |
+
tags:
|
| 6 |
+
- image-to-image
|
| 7 |
+
- watermark-removal
|
| 8 |
+
- remove-watermark
|
| 9 |
+
- watermark
|
| 10 |
+
- torchscript
|
| 11 |
+
- computer-vision
|
| 12 |
+
- image-processing
|
| 13 |
+
- image-restoration
|
| 14 |
+
- image-cleaning
|
| 15 |
+
pipeline_tag: image-to-image
|
| 16 |
+
library_name: pytorch
|
| 17 |
+
---
|
| 18 |
+
|
| 19 |
# Fast Watermark Removal
|
| 20 |
|
| 21 |
A high-performance TorchScript model for removing watermarks from images. This model uses a dual-stage architecture optimized for speed and quality.
|
| 22 |
|
| 23 |
+
## Test the Model
|
| 24 |
+
|
| 25 |
+
Try the model instantly in your browser — no setup required:
|
| 26 |
+
|
| 27 |
+
**[Remove Watermarks → clearpics.ai](https://clearpics.ai/remove-watermarks)**
|
| 28 |
+
|
| 29 |
## Features
|
| 30 |
|
| 31 |
- **Fast inference**: ~500ms per image (RTX 4090)
|
|
|
|
| 33 |
- **Production-ready**: Compiled TorchScript model, no training code needed
|
| 34 |
- **Memory efficient**: Requires 11.5GB VRAM
|
| 35 |
|
| 36 |
+
## Technical Details
|
| 37 |
+
|
| 38 |
+
- **Architecture**: Dual-stage with Swin2 Transformers
|
| 39 |
+
- **Format**: TorchScript (.ts) compiled model
|
| 40 |
+
- **Input**: RGB images (any resolution)
|
| 41 |
+
- **Output**: RGB images (max 768px, aspect ratio preserved)
|
| 42 |
+
- **Precision**: FP32 with TensorFloat32 matmul on Ampere+ GPUs
|
| 43 |
+
- **Batch size**: 1
|
| 44 |
+
|
| 45 |
## Limitations
|
| 46 |
|
| 47 |
- **Output resolution**: Limited to 768px maximum dimension (aspect ratio preserved)
|
|
|
|
| 67 |
### Setup
|
| 68 |
|
| 69 |
```bash
|
|
|
|
|
|
|
|
|
|
|
|
|
| 70 |
# Install dependencies
|
| 71 |
pip install -r requirements.txt
|
| 72 |
```
|
|
|
|
| 82 |
### Batch Processing
|
| 83 |
|
| 84 |
```bash
|
| 85 |
+
python inference.py -f /path/to/images/folder -m model.ts
|
| 86 |
```
|
| 87 |
|
| 88 |
### Arguments
|
|
|
|
| 111 |
|
| 112 |
All processing (including resizing and normalization) is performed within the compiled TorchScript model for optimal performance.
|
| 113 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 114 |
## Future Improvements
|
| 115 |
|
| 116 |
I'm actively exploring ways to enhance this model's capabilities. If you have suggestions, encounter issues, or are interested in collaborating on improvements, please reach out!
|
| 117 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 118 |
|
| 119 |
## License
|
| 120 |
|