env instractions updated
Browse files- README.md +18 -6
- environment.yml +4 -0
- requirements.txt +15 -0
README.md
CHANGED
|
@@ -1,10 +1,11 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
language:
|
| 4 |
-
- en
|
| 5 |
tags:
|
| 6 |
-
- image-restoration
|
| 7 |
---
|
|
|
|
| 8 |
<table>
|
| 9 |
<tr>
|
| 10 |
<td align="center">Input (B&W)</td>
|
|
@@ -32,7 +33,7 @@ This application provides a web-based interface for colorizing black and white i
|
|
| 32 |
|
| 33 |
### Prerequisites
|
| 34 |
|
| 35 |
-
- Python 3.
|
| 36 |
- CUDA-capable GPU (recommended for faster processing)
|
| 37 |
- Git, Git-Lfs
|
| 38 |
|
|
@@ -52,14 +53,24 @@ You can set up the environment using either Conda (recommended) or pip.
|
|
| 52 |
|
| 53 |
```bash
|
| 54 |
conda env create -f environment.yml
|
| 55 |
-
conda activate deoldify
|
| 56 |
```
|
| 57 |
|
| 58 |
- **Using Pip:**
|
| 59 |
|
| 60 |
-
Make sure you have a compatible Python environment (3.
|
| 61 |
|
| 62 |
```bash
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
pip install -r requirements.txt
|
| 64 |
```
|
| 65 |
|
|
@@ -130,7 +141,8 @@ The web application provides an intuitive interface for:
|
|
| 130 |
- If you encounter CUDA out-of-memory errors, try lowering the render factor
|
| 131 |
- For best results with the artistic model, a render factor of 10-15 is recommended
|
| 132 |
- For the stable model, a render factor of 20-25 typically works well
|
|
|
|
| 133 |
|
| 134 |
## License
|
| 135 |
|
| 136 |
-
This project uses DeOldify, which is licensed under MIT License.
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
language:
|
| 4 |
+
- en
|
| 5 |
tags:
|
| 6 |
+
- image-restoration
|
| 7 |
---
|
| 8 |
+
|
| 9 |
<table>
|
| 10 |
<tr>
|
| 11 |
<td align="center">Input (B&W)</td>
|
|
|
|
| 33 |
|
| 34 |
### Prerequisites
|
| 35 |
|
| 36 |
+
- Python 3.10
|
| 37 |
- CUDA-capable GPU (recommended for faster processing)
|
| 38 |
- Git, Git-Lfs
|
| 39 |
|
|
|
|
| 53 |
|
| 54 |
```bash
|
| 55 |
conda env create -f environment.yml
|
| 56 |
+
conda activate deoldify
|
| 57 |
```
|
| 58 |
|
| 59 |
- **Using Pip:**
|
| 60 |
|
| 61 |
+
Make sure you have a compatible Python environment (3.10).
|
| 62 |
|
| 63 |
```bash
|
| 64 |
+
# Create a virtual environment
|
| 65 |
+
python -m venv .venv
|
| 66 |
+
|
| 67 |
+
# Activate the virtual environment
|
| 68 |
+
# On Windows:
|
| 69 |
+
.venv\Scripts\activate
|
| 70 |
+
# On macOS/Linux:
|
| 71 |
+
source .venv/bin/activate
|
| 72 |
+
|
| 73 |
+
# Install the required dependencies
|
| 74 |
pip install -r requirements.txt
|
| 75 |
```
|
| 76 |
|
|
|
|
| 141 |
- If you encounter CUDA out-of-memory errors, try lowering the render factor
|
| 142 |
- For best results with the artistic model, a render factor of 10-15 is recommended
|
| 143 |
- For the stable model, a render factor of 20-25 typically works well
|
| 144 |
+
- increasing the render factor consume more gpu memory.
|
| 145 |
|
| 146 |
## License
|
| 147 |
|
| 148 |
+
This project uses DeOldify, which is licensed under MIT License.
|
environment.yml
CHANGED
|
@@ -21,3 +21,7 @@ dependencies:
|
|
| 21 |
- opencv-python>=4.2.0.32
|
| 22 |
- wandb
|
| 23 |
- yt-dlp
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
- opencv-python>=4.2.0.32
|
| 22 |
- wandb
|
| 23 |
- yt-dlp
|
| 24 |
+
- fastapi==0.104.0
|
| 25 |
+
- uvicorn==0.23.2
|
| 26 |
+
- streamlit==1.27.0
|
| 27 |
+
- python-multipart==0.0.6
|
requirements.txt
CHANGED
|
@@ -1,3 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
fastapi==0.104.0
|
| 2 |
uvicorn==0.23.2
|
| 3 |
streamlit==1.27.0
|
|
|
|
| 1 |
+
# -----------------------------------------------------------------------------
|
| 2 |
+
# DeOldify Requirements
|
| 3 |
+
#
|
| 4 |
+
# This file lists the pip-installable dependencies for the DeOldify project.
|
| 5 |
+
#
|
| 6 |
+
# Note:
|
| 7 |
+
# - This requirements file was derived from a Conda environment YAML file.
|
| 8 |
+
# - Some system-level or non-Python dependencies (e.g., mkl, ffmpeg) are
|
| 9 |
+
# not listed here and must be installed separately.
|
| 10 |
+
# - Please ensure you are using Python 3.10 as specified in the original spec.
|
| 11 |
+
#
|
| 12 |
+
# - or just install conda and run `conda env create -f environment.yml`
|
| 13 |
+
# -----------------------------------------------------------------------------
|
| 14 |
+
|
| 15 |
+
|
| 16 |
fastapi==0.104.0
|
| 17 |
uvicorn==0.23.2
|
| 18 |
streamlit==1.27.0
|