Unverified Commit 3347abea authored by narugo1992's avatar narugo1992 Committed by GitHub
Browse files

Merge pull request #31 from deepghs/dev/nsfw 

dev(narugo): add nsfw prediction from nsfwjs library in javascript
parents 5b318d1e c671c376

docs/source/api_doc/validate/index.rst

+1 −0
Original line number Diff line number Diff line
@@ -13,5 +13,6 @@ imgutils.validate 
    classify

    color

    monochrome

    nsfw

    rating

    truncate

docs/source/api_doc/validate/nsfw.plot.py

0 → 100644
+16 −0
Original line number Diff line number Diff line 
import glob

import os.path



from natsort import natsorted



from plot import image_plot



if __name__ == '__main__':

    image_plot(

        *natsorted(glob.glob(os.path.join('nsfw', 'drawings', '*.jpg'))),

        *natsorted(glob.glob(os.path.join('nsfw', 'hentai', '*.jpg'))),

        *natsorted(glob.glob(os.path.join('nsfw', 'neutral', '*.jpg'))),

        *natsorted(glob.glob(os.path.join('nsfw', 'porn', '*.jpg'))),

        *natsorted(glob.glob(os.path.join('nsfw', 'sexy', '*.jpg'))),

        columns=4, figsize=(10, 18.5),

    )

docs/source/api_doc/validate/nsfw.plot.py.svg

0 → 100644
+1497 −0

File added.

Preview size limit exceeded, changes collapsed.

docs/source/api_doc/validate/nsfw.rst

0 → 100644
+22 −0
Original line number Diff line number Diff line 
imgutils.validate.nsfw

=============================================



.. currentmodule:: imgutils.validate.nsfw



.. automodule:: imgutils.validate.nsfw





nsfw_pred_score

----------------------------------------



.. autofunction:: nsfw_pred_score







nsfw_pred

----------------------------------------



.. autofunction:: nsfw_pred

docs/source/api_doc/validate/nsfw/drawings/1.jpg

0 → 100644
+28.3 KiB
Loading image diff...