0.10.1

• Remove grad_cam_height and grad_cam_width settings, the output grad-cam image will always be the same size as the input image
• Default alpha is 0.8

0.9.2

• Add progress bar for predicting video frames.
• Set all default batch sizes to 8.

0.8.0

Add aggregation support for predicting video frames.

0.7.0

• Drop support for Python 3.7.
• Use numpy.typing from NumPy 1.22.0.
• Use Pillow 9.0.0, note that this leads to some small floating point differences to the results.
• Fix Grad-CAM tools.

0.6.1

Add matplotlib to required packages.

0.6.0

New features:

• predict_image and predict_images can now also produce Grad-CAM plots that highlight the important regions which lead to the predictions.

0.5.3

Clean up code for model architecture to improve readability.

0.5.2

Fix mypy re-export errors

0.5.1

Save output video, if enabled, as MP4 instead of AVI.

0.5.0

New:

• predict_video_frames for getting the NSFW probability of each video frame
• predict_image for getting the NSFW probability of a single image

Breaking changes:

• predict is removed, and replaced by predict_images, which returns a list of NSFW probabilities of the input images