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.
The threshold can also be overridden on a per-request basis with the
threshold query parameter (a value between 0 and 1 inclusive):
curl -X POST -F "image=@image.png" "https://deepdanbooru.nsk.sh/deepdanbooru?threshold=0.2"
A threshold of 0 returns scores for all tags.