Keras implementation of the Yahoo Open-NSFW model
MIT License
Bot releases are visible (Hide)
Published by bhky 3 months ago
PIL.Image.Image
.global_average_pooling2d
for the GlobalAveragePooling2D
layer, to avoid Keras assigning a different name upon multiple model instantiations.Published by bhky 10 months ago
KerasTensor
is defined instead of using keras.KerasTensor
directly, as the latter is only available in Keras 3.Published by bhky 10 months ago
name
to the make_open_nsfw_model
function. This will be used as the name
attribute of the returned Keras model object. The default name is opennsfw2
.Published by bhky 10 months ago
Published by bhky 11 months ago
Published by bhky 11 months ago
Published by bhky about 1 year ago
tf.keras
to keras_core
Published by bhky over 2 years ago
grad_cam_height
and grad_cam_width
settings, the output grad-cam image will always be the same size as the input imagealpha
is 0.8Published by bhky over 2 years ago
Published by bhky over 2 years ago
Add aggregation support for predicting video frames.
Published by bhky almost 3 years ago
numpy.typing
from NumPy 1.22.0
.9.0.0
, note that this leads to some small floating point differences to the results.Published by bhky almost 3 years ago
Add matplotlib
to required packages.
Published by bhky almost 3 years ago
Clean up code for model architecture to improve readability.
Published by bhky almost 3 years ago
Fix mypy re-export errors
Published by bhky almost 3 years ago
Save output video, if enabled, as MP4 instead of AVI.
Published by bhky almost 3 years ago
New:
predict_video_frames
for getting the NSFW probability of each video framepredict_image
for getting the NSFW probability of a single imageBreaking changes:
predict
is removed, and replaced by predict_images
, which returns a list of NSFW probabilities of the input imagesPublished by bhky almost 3 years ago
Update dependencies.