Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import os
- from nudenet import NudeDetector
- detector = NudeDetector()
- directory = 'image path in'
- directory_out = 'image path out'
- #for now it only works with a path that doesn't contain subfolders
- loop= [os.path.join(directory, file) for file in os.listdir(directory)]
- loop_out= [os.path.join(directory_out, file) for file in os.listdir(directory)]
- ptb=['EXPOSED_ANUS','EXPOSED_BUTTOCKS','EXPOSED_BREAST_F','EXPOSED_GENITALIA_F','COVERED_BREAST_F']
- for i in range(0,len(loop)): detector.censor(
- loop[i],
- out_path=loop_out[i],
- visualize=False,
- parts_to_blur=ptb,
- typec='pix',
- pixs=[8,8]
- )
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement