From 114a8df2288464e44d2e96afd4144387e9db06a3 Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Sat, 30 Jan 2021 00:15:18 -0300 Subject: [PATCH] change readme --- README.md | 24 +++++++++++++++++++++++- 1 file changed, 23 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 28df45b..e6b1c34 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,5 @@ + + # Rembg [![Downloads](https://pepy.tech/badge/rembg)](https://pepy.tech/project/rembg) @@ -38,6 +40,26 @@ Rembg is a tool to remove images background. That is it. * python 3.8 or newer +* torch and torchvision stable version (https://pytorch.org) + +====> How to install torch/torchvision <==== + +Go to https://pytorch.org and scrool down to `INSTALL PYTORCH` section and follow the instructions. + +For example: +``` +PyTorch Build: Stable (1.7.1) +Your OS: Windows +Package: Pip +Language: Python +CUDA: None +``` + +The install cmd is: +``` +pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html +``` + ### Installation Install it from pypi @@ -99,7 +121,7 @@ Then run ``` cat input.png | python app.py > out.png ``` - + #### Example 2: Using PIL In `app.py`