diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..85179d8 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,7 @@ +FROM nvidia/cuda + +RUN apt-get update && apt-get install -y --no-install-recommends python3 python3-pip python3-dev llvm llvm-dev +RUN pip3 install rembg + +ENTRYPOINT ["rembg"] +CMD [] diff --git a/README.md b/README.md index c341743..c39facb 100644 --- a/README.md +++ b/README.md @@ -99,6 +99,14 @@ Then run cat input.png | python app.py > out.png ``` +### Usage as a docker + +Just run + +``` + curl -s http://input.png | docker run -i -v ~/.u2net:/root/.u2net danielgatis/rembg:latest > output.png +``` + ### Advance usage Sometimes it is possible to achieve better results by turning on alpha matting. Example: