Framework Inference
For NSFW models, the inference script generates image scores for NSFW content. Values closer to 1 are strong NSFW while -1 are strong safe.
To enable inference stage with a NSFW model, configure the configuration files:
In the
defaults
section ofconf/config.yaml
, update thefw_inference
field to point to the desired NSFW inference configuration file. For example, if you want to use thensfw/nsfw.yaml
configuration, change thefw_inference
field tonsfw/nsfw
.defaults: - fw_inference: nsfw/nsfw ...
In the
stages
field ofconf/config.yaml
, make sure thefw_inference
stage is included. For example,stages: - fw_inference ...
Configure
image_path
field ofconf/fw_inference/nsfw/nsfw.yaml
.Execute launcher pipeline:
python3 main.py