跳到内容

夜深了,注意休息,愿你今夜好梦。

  • 2 回复
  • 194 浏览

求个能批量给3万张图打标签的本地工具,预算卡死两千

参与讨论

刚刷小红书「一人说一个信息差」,评论区全在吵效率工具,我这就一个死任务:三万多张商品图,按颜色和品类打标签,得在本地跑,数据不能出门。
机器是3060 12G,试过几个开源方案,一张图要跑两秒往上,全跑完得一天多。两千以内有没有能压到0.5秒的,或者我参数没调对。

讨论区

按楼层回复,支持引用与表情;使用 @用户名 或 @昵称 可提醒对方;发言计入圈子贡献字数。

2s 条评论

  1. 周报难产中
    周报难产中 成长7,820

    3060 12G 跑 3 万张要压到 0.5 秒,基本得换小模型加批处理,先把图像解码那步用 DALI 或 OpenCV 多线程喂满,别让 GPU 干等。预算两千内可以看 ONNX Runtime 加 CLIP 蒸馏小模型,标签固定的话直接训个轻量分类头,比通用打标快一截。@周末补作业 你这标签是固定几十类还是开放式描述,差别挺大。

  2. 细节控发作
    细节控发作 萌芽1,216
    引用 周报难产中

    3060 12G 跑 3 万张要压到 0.5 秒,基本得换小模型加批处理,先把图像解码那步用 DALI 或 OpenCV 多线程喂满,别让 GPU 干等。预算两千内可以看 ONNX Runtime 加 CLIP 蒸馏小模型,标签固定的话直接训个轻量分类头,比通用打标快一截。@周末补作业 你这标签是固定几十类还是开放式描…

    @周末补作业 标签要是固定几十类,别上通用打标模型,直接拿 CLIP 蒸馏个小分类头 + ONNX Runtime,batch 开到 32 往上,3060 压 0.5 秒有戏。预算两千内够,就是得自己标个几百张做微调,纯开箱即用的工具没这速度。

发表评论

登录后发表评论

登录即可参与楼层讨论,支持引用回复与 @ 提醒。

前往登录页

Welcome! This site is in Chinese. Tap EN in the top bar to read in English.