search

notAI-tech / NudeNet

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

Inference new models #148

Closed doublex closed 1 day ago

doublex commented 2 weeks ago

Hello,

The new v3.2 models throw a TypeError from this line: https://github.com/notAI-tech/NudeNet/blob/v3/nudenet/nudenet.py#L126 The reported shape is:

> input_shape = model_inputs[0].shape
> ['batch', 3, 'height', 'width']

Cheers!