Update README.md
Browse files
README.md
CHANGED
|
@@ -17,10 +17,9 @@ This repository provides an integrated set of image-processing modules aimed at
|
|
| 17 |
|
| 18 |
**Usage**:
|
| 19 |
- Install the required libraries:
|
| 20 |
-
```bash
|
| 21 |
pip install numpy opencv-python
|
| 22 |
|
| 23 |
-
|
| 24 |
1) specify the input_folder that has the faceimages and the output_folder to get the composite sketches
|
| 25 |
2) python cat.py
|
| 26 |
|
|
@@ -31,42 +30,34 @@ Adjust gamma in the adjust_gamma() function for brightness/contrast fine-tuning.
|
|
| 31 |
|
| 32 |
Description: A GAN-based module implemented in TensorFlow, trained specifically to restore blurred facial sketches. The module leverages a deep convolutional generator and a patch-based discriminator.
|
| 33 |
|
| 34 |
-
Key Features:
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
Loss Function: Optimized using Wasserstein loss for stability during training.
|
| 39 |
-
|
| 40 |
-
Usage:
|
| 41 |
-
|
| 42 |
-
Install TensorFlow and required dependencies:
|
| 43 |
|
| 44 |
-
|
|
|
|
|
|
|
| 45 |
|
| 46 |
-
**Steps to
|
| 47 |
1. Prepare Input data: Place the folder of clear sketches in the root directory of this project. Rename it to clear_sketches
|
| 48 |
-
2. Run the blur code to
|
| 49 |
3. python deblurring.py
|
| 50 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
### **3. Image Translator Module**
|
| 53 |
|
| 54 |
-
Description: Converts sketches into photo-realistic images using a
|
| 55 |
-
|
| 56 |
-
Usage:
|
| 57 |
-
|
| 58 |
-
Install PyTorch and necessary libraries:
|
| 59 |
-
|
| 60 |
-
pip install torch torchvision
|
| 61 |
-
|
| 62 |
-
Train or test the image translator model:
|
| 63 |
-
|
| 64 |
-
python image_translator.py
|
| 65 |
|
| 66 |
-
|
|
|
|
|
|
|
| 67 |
|
| 68 |
-
Python 3.x
|
| 69 |
-
Libraries: TensorFlow, PyTorch, NumPy, OpenCV
|
| 70 |
|
| 71 |
Acknowledgments
|
| 72 |
|
|
|
|
| 17 |
|
| 18 |
**Usage**:
|
| 19 |
- Install the required libraries:
|
|
|
|
| 20 |
pip install numpy opencv-python
|
| 21 |
|
| 22 |
+
**Steps to run the script:**
|
| 23 |
1) specify the input_folder that has the faceimages and the output_folder to get the composite sketches
|
| 24 |
2) python cat.py
|
| 25 |
|
|
|
|
| 30 |
|
| 31 |
Description: A GAN-based module implemented in TensorFlow, trained specifically to restore blurred facial sketches. The module leverages a deep convolutional generator and a patch-based discriminator.
|
| 32 |
|
| 33 |
+
**Key Features**:
|
| 34 |
+
- **Generator**: Transforms blurred images into sharp images using a combination of convolutional, residual, and transposed convolutional layers.
|
| 35 |
+
- **Discriminator**: Employs convolutional layers to distinguish real and synthetic sketches.
|
| 36 |
+
- **Loss Function:** Optimized using Wasserstein loss for stability during training.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
|
| 38 |
+
**Usage:**
|
| 39 |
+
- Install TensorFlow and required dependencies:
|
| 40 |
+
pip install tensorflow
|
| 41 |
|
| 42 |
+
**Steps to Train:**
|
| 43 |
1. Prepare Input data: Place the folder of clear sketches in the root directory of this project. Rename it to clear_sketches
|
| 44 |
+
2. Run the blur code to introduce artificial blur (You can choose any blurring technique ).
|
| 45 |
3. python deblurring.py
|
| 46 |
|
| 47 |
+
**Steps for Inference:**
|
| 48 |
+
1. place your Blurred sketches in the folder names blurred_input (.jpg, .png )
|
| 49 |
+
2. python deblur_inference.py
|
| 50 |
+
3. Deblurred images are saved in a folder named deblurred_output.
|
| 51 |
+
4. The filenames will be prefixed with deblurred_.
|
| 52 |
|
| 53 |
### **3. Image Translator Module**
|
| 54 |
|
| 55 |
+
Description: Converts sketches into photo-realistic images using a GAN architecture, preserving structure and enhancing realism.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
|
| 57 |
+
**Usage:**
|
| 58 |
+
- Install TensorFlow and required dependencies:
|
| 59 |
+
pip install tensorflow
|
| 60 |
|
|
|
|
|
|
|
| 61 |
|
| 62 |
Acknowledgments
|
| 63 |
|