mirror
/
MODNet
mirror of https://github.com/ZHKKKe/MODNet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update the requirements.txt for the webcam demo

pull/35/head
ZHKKKe 2020-12-22 00:41:22 +08:00
parent fea4e3a2ee
commit 647803e76d
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
demo/video_matting/webcam/README.md
View File

@ -32,13 +32,13 @@ We recommend creating a new conda virtual environment to run this demo, as follo
2. Download the pre-trained model from this [link](https://drive.google.com/file/d/1Nf1ZxeJZJL8Qx9KadcYYyEmmlKhTADxX/view?usp=sharing) and put it into the folder `MODNet/pretrained/`. 2. Download the pre-trained model from this [link](https://drive.google.com/file/d/1Nf1ZxeJZJL8Qx9KadcYYyEmmlKhTADxX/view?usp=sharing) and put it into the folder `MODNet/pretrained/`.
3. Create a conda virtual environment named `modnet` (if it doesn't exist) and activate it: 3. Create a conda virtual environment named `modnet` (if it doesn't exist) and activate it. Here we use `python=3.6` as an example:
``` ```
conda create -n modnet python=3.6 conda create -n modnet python=3.6
source activate modnet source activate modnet
``` ```
4. Install the required python dependencies (here we use PyTorch==1.0.0): 4. Install the required python dependencies (please make sure your CUDA version is supported by the PyTorch version installed):
``` ```
pip install -r demo/video_matting/webcam/requirements.txt pip install -r demo/video_matting/webcam/requirements.txt
``` ```

2
demo/video_matting/webcam/requirements.txt
View File

@ -1,5 +1,5 @@
numpy numpy
Pillow Pillow
opencv-python opencv-python
torch == 1.0.0 torch >= 1.0.0
torchvision torchvision