jasonengage Claude Sonnet 4.6 commited on
Commit
fdaa122
·
1 Parent(s): b5e53f5

Updated README with metadata, tags, and test link

Browse files

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Files changed (1) hide show
  1. README.md +34 -19
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 -o output_folder
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