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

pretrained model best.onnx #124

Closed waquey closed 8 months ago

waquey commented 9 months ago

Dear great author, Thanks for the great work. I've tried to view the "best.onnx" and find it is converted from pytorch 2.0.1 with onnx v8. Could you please provide the original pytorch model graph as well?

Currently I've tried to use pytorch (torch.nn.Conv2d(128, 64, kernel_size=(1, 1), stride=(1, 1) )) to build graph, which is /model.4/cv2/conv/Conv_output_0 [kernel_shape = stride = 1 and pads=0,0,0,0] in onnx

However, the results vary a lot for pytorch and onnx even if the weights/bias is the same

Thanks

bedapudi6788 commented 8 months ago

@waquey https://github.com/notAI-tech/NudeNet/blob/v3/best.pt is the pytorch model.