oofdere / deepdanbooru-docker

Docker service for https://github.com/KichangKim/DeepDanbooru

deepdanbooru-docker

Docker Hub | Demo

Docker service for https://github.com/KichangKim/DeepDanbooru.

This uses the deepdanbooru-onnx library (GitHub repository) which has converted the original TensorFlow model into an ONNX model.

Installation

docker run --publish $PORT:$PORT --env PORT=$PORT --detach nanoskript/deepdanbooru

Configuration

Threshold

The confidence threshold for which results are filtered by can be configured with the DEEPDANBOORU_THRESHOLD environment variable:

docker run --env DEEPDANBOORU_THRESHOLD=0.0 ... 

By default, this is set to 0.5.