织图AI

织图AI

首页
商品及服务
Apps
瞬间
教程
关注
登录 →
织图AI

织图AI

首页 商品及服务 Apps 瞬间 教程 关注
登录
  1. 首页
  2. 人工智能
  3. 我刚开源了一个图片NSFW(鉴黄)推理模型!毫秒级,仅23.7 MB,能够用 onnxruntime 在服务器上运行

我刚开源了一个图片NSFW(鉴黄)推理模型!毫秒级,仅23.7 MB,能够用 onnxruntime 在服务器上运行

0
  • 人工智能
  • 发布于 2024-10-24
  • 81 次阅读
织图AI
织图AI

首先我不是介绍别人开源项目的公众号,我介绍的是我们系统开源的项目!有需要的可以看看!

我们造轮子,给大家用,就像我们的口号,让科技有温度,用技术惠大众!

咳咳,我一个小瘪三,上面的 slogan 让我老脸一红,一个小瘪三也大言不惭,哈哈哈 😂😂😂

言归正传

本来是给自家系统图片合规用的,说白了就是用来鉴定用户使用或上传的是不是有颜色的图片,虽然市场上也有付费的服务,但多多少少都得花钱,有句话说得好赚3块省2块得5块!

主要是自己也在研究AI,研究人工智能做这个也花了不少心思,主要是自己能力不够,只能勤能补拙,对,就是堆时间!

创业跌跌撞撞,深懂里面的苦涩与不易,这是我们系统里AI组件的一部分,索性就直接开源。你也可以把它放在系统里当一个插件用。也能节省点人工审核的成本,少操点不老实用户的心。

1.推理也是毫秒级的(用了单例设计?的原因??)

2.准确率高

3.模型只有23.7MB,非常轻量化,放在服务器里十分节省资源

4.使用 onnxruntime 服务器使用起来很方便

插播个自己做的小程序!

txl_15.jpg

简单介绍下用法

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

更多内容请关注公众号

jy_8.jpg

标签: #大模型 1
相关文章
我刚开源了一个图片NSFW(鉴黄)推理模型!毫秒级,仅23.7 MB,能够用 onnxruntime 在服务器上运行

我刚开源了一个图片NSFW(鉴黄)推理模型!毫秒级,仅23.7 MB,能够用 onnxruntime 在服务器上运行 2024-10-24 03:20

首先我不是介绍别人开源项目的公众号,我介绍的是我们系统开源的项目!有需要的可以看看! 我们造轮子,给大家用,就像我们的口号,让科技有温度,用技术惠大众! 咳咳,我一个小瘪三,上面的 slogan 让我老脸一红,一个小瘪三也大言不惭,哈哈哈 😂😂😂

目录
  • 加入qq群
Copyright © 2024 All Rights Reserved. Powered by 惊鱼科技.
发邮件 bu@stunningfish.com
沪ICP备2023010671号-4