Docker Hub | API documentation | Web tool
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.
docker run --publish $PORT:$PORT --env PORT=$PORT --detach nanoskript/deepdanbooru
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.