首先我不是介绍别人开源项目的公众号,我介绍的是我们系统开源的项目!有需要的可以看看!
我们造轮子,给大家用,就像我们的口号,让科技有温度,用技术惠大众!
咳咳,我一个小瘪三,上面的 slogan 让我老脸一红,一个小瘪三也大言不惭,哈哈哈 😂😂😂
言归正传
本来是给自家系统图片合规用的,说白了就是用来鉴定用户使用或上传的是不是有颜色的图片,虽然市场上也有付费的服务,但多多少少都得花钱,有句话说得好赚3块省2块得5块!
主要是自己也在研究AI,研究人工智能做这个也花了不少心思,主要是自己能力不够,只能勤能补拙,对,就是堆时间!
创业跌跌撞撞,深懂里面的苦涩与不易,这是我们系统里AI组件的一部分,索性就直接开源。你也可以把它放在系统里当一个插件用。也能节省点人工审核的成本,少操点不老实用户的心。
1.推理也是毫秒级的(用了单例设计?的原因??)
2.准确率高
3.模型只有23.7MB,非常轻量化,放在服务器里十分节省资源
4.使用 onnxruntime 服务器使用起来很方便
插播个自己做的小程序!
简单介绍下用法
from nsfw.nsfw_service import setup_nsfw
classifier = setup_nsfw() # 使用默认模型[完全够用]
#classifier = setup_nsfw('you.onnx') # 使用自定义模型
# 测试图片路径image = '/path/to/image.jpg'
result = classifier.nsfw_risk_ndh(image)
result2 = classifier.nsfw_risk_tf(image)
print(f"result image: {result}")
print(f"result image2: {result2}")
print("-" * 40)
返回结果:
result image: 0.7988816380500793
result image2: True
----------------------------------------
🐱github地址:
https://github.com/JingYuTech/nsfw
😊huggingface地址:
https://huggingface.co/starlette/img_nsfw
更多内容请关注公众号