Onnx Post-Processing, NMS, TensoRT - Githubissues

notAI-tech / NudeNet

Lightweight nudity detection

https://nudenet.notai.tech/

GNU Affero General Public License v3.0

1.66k stars 335 forks source link

Onnx Post-Processing, NMS, TensoRT #127

Open ChuRuaNh0 opened 9 months ago

ChuRuaNh0 commented 9 months ago

I had a repo to add nms into NudeNet onnx and generate tensorrt with inferences. You can refer: https://github.com/ChuRuaNh0/NudeNet-Onnx-TensorRT-BatchedNMS

bedapudi6788 commented 8 months ago

@ChuRuaNh0 It looks good. Can I copy from this and add to nudenet repo?

ChuRuaNh0 commented 8 months ago

@bedapudi6788 Yes, Welcome.

ChuRuaNh0 commented 8 months ago

@bedapudi6788 Can you add me in contributor or Readme, I'm going to appreciate.

bedapudi6788 commented 8 months ago

@ChuRuaNh0 Hey, for sure, I will. Didnt get the time to add your changes to the repo yet. bit busy for the next couple of weeks. I will do so after.

meanwhile, if you wanna submit a PR with readme and code changes, feel free to.