diff --git a/bmf/demo/colorization_python/Readme.md b/bmf/demo/colorization_python/Readme.md index b4cebbe4..8e982cdd 100644 --- a/bmf/demo/colorization_python/Readme.md +++ b/bmf/demo/colorization_python/Readme.md @@ -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 @@ -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 @@ -73,7 +73,7 @@ ffmpeg -version ```Python import sys -sys.path.insert(0, '/content/DeOldify') +sys.path.insert(0, './DeOldify') print(sys.path) ``` diff --git a/bmf/demo/colorization_python/requirements.txt b/bmf/demo/colorization_python/requirements.txt new file mode 100644 index 00000000..b56cdb9d --- /dev/null +++ b/bmf/demo/colorization_python/requirements.txt @@ -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 \ No newline at end of file