Skip to content

Commit

Permalink
Fix the bug of Missing dependencies and DeOldify path set in demo col…
Browse files Browse the repository at this point in the history
…orization_python (#118)

* fix the bug of requirements and DeOldify path set
* Update Readme.md
  • Loading branch information
JackLau1222 authored Sep 10, 2024
1 parent 17aee9d commit d3b61cc
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 3 deletions.
6 changes: 3 additions & 3 deletions bmf/demo/colorization_python/Readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ git clone https://github.com/eefengwei/DeOldify.git DeOldify

### 1.2 install the dependent python packages
```Bash
pip3 install -r ./DeOldify/requirements-colab.txt
pip3 install -r ./requirements.txt
```

### 1.3 download the pretrained-weights
Expand All @@ -46,7 +46,7 @@ wget -c https://huggingface.co/spensercai/DeOldify/resolve/main/ColorizeVideo_ge

### 1.4 pip install BMF packages
```Bash
pip3 install bmf
pip3 install BabitMF-GPU
```
### 1.5 verify the FFmpeg libraries is installed and version is correct

Expand All @@ -73,7 +73,7 @@ ffmpeg -version

```Python
import sys
sys.path.insert(0, '/content/DeOldify')
sys.path.insert(0, './DeOldify')
print(sys.path)
```

Expand Down
9 changes: 9 additions & 0 deletions bmf/demo/colorization_python/requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
fastai==1.0.60
tensorboardX>=1.6
ffmpeg-python
yt-dlp
opencv-python>=4.2.0.32
Pillow
tornado
imgaug==0.2.6
ipython

0 comments on commit d3b61cc

Please sign in to comment.