search

notAI-tech / NudeNet

Lightweight nudity detection
https://nudenet.notai.tech/
GNU Affero General Public License v3.0
1.76k stars 342 forks source link

Has the docker-version gpu support? #85

Open MrTsunami opened 3 years ago

MrTsunami commented 3 years ago

Is it possible to run the docker-version of the detector with gpu support? Also, is it possible to enable fast-mode within docker?

bedapudi6788 commented 3 years ago

I will upload new docker images supporting bot of these in an hour.

edit: misread your issue to be regarding classifier gpu support. current detector doesn't work on gpu (https://github.com/notAI-tech/NudeNet/issues/84) with an onnxruntime error. You will have to build a docker image with tensorflow versions of the detector (check previous releases).

MrTsunami commented 3 years ago

Thank you for your dedication!

What about the fast mode? Is it useful to run multiple Docker instances on a multicore cpu or does Nudenet already use multiple cores by itself?

bedapudi6788 commented 3 years ago

@MrTsunami I pushed a new docker image for detector that will use the fast model if the image name has _fast_mode_ in the name. https://github.com/notAI-tech/NudeNet/blob/af4d8e4634a5d4cc5b90ad9ce5f633d56facd5b5/fastDeploy_recipes/detector/predictor.py#L23