Browse Source

add dockerfile

Daniel Gatis 4 years ago
parent
commit
e40b1e3e6b
2 changed files with 15 additions and 0 deletions
  1. 7 0
      Dockerfile
  2. 8 0
      README.md

+ 7 - 0
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 []

+ 8 - 0
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: