backgroundremover test

docker-compose.yml

version: '3.3'
services:
  backgroundremover:
    build:
      context: ./
      dockerfile: Dockerfile
      args:
        - UID=1000
        - GID=1001
    container_name: "backgroundremover"
    user: "1000:1001"
    restart: "no"
    volumes:
      - './images:/app:rw'
    working_dir: /app

Dockerfile

FROM python:3.6-slim

ARG UID
ARG GID

RUN groupadd -g "${GID}" app \
    && useradd --create-home --no-log-init -u "${UID}" -g "${GID}" app

RUN apt-get update && \
	apt-get -y install python3-pip && \
	apt-get -y install ffmpeg imagemagick

USER app

RUN pip install pip
RUN pip install --default-timeout=1000 torch torchvision backgroundremover

WORKDIR /app

ENTRYPOINT ["tail"]
CMD ["-f","/dev/null"]

use.txt

docker exec -it backgroundremover /bin/bash

/home/app/.local/bin/backgroundremover -i input.jpeg -m u2net_human_seg -o output.jpeg

convert output.jpeg -colorspace Gray -trim final.png