@@ -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 []
@@ -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: