
transformers 三行代码跑通推理与目标检测【免费下载链接】transformers Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.项目地址: https://gitcode.com/GitHub_Trending/tra/transformers想让应用看懂一张照片或给业务接上一个现成模型你多半会被同一件事劝退换个模型预处理、加载、后处理就得跟着重写一遍。transformers 是 Hugging Face 推出的模型定义框架把推理和训练收敛到同一套 API 上——你负责喂数据和收结果剩下的交给它。能力速览一张表看懂它能干什么一句话定义transformers 统一了模型怎么定义这件事训练框架DeepSpeed、FSDP和推理引擎vLLM、SGLang都直接消费它的模型定义一份定义全生态通用。能力解决什么问题对应模块一行 Pipeline 推理不用手写预处理和后处理pipelines/Auto 系列自动加载不手写模型结构代码models/文本生成对话与文案创作text-generation目标检测识别图中物体及其位置object-detectionTrainer 微调预训练模型适配自有数据trainer.py这是仓库测试目录里的一张 COCO 样本图。两只猫、两个遥控器都会被检测 pipeline 框出来并标上类别。最小起步安装 transformers 并跑通第一次文本生成 先装库方括号里的 torch 表示同时装上 PyTorch 后端pip install transformers[torch]然后调用 pipeline整个推理入口就一个构造函数from transformers import pipeline pipe pipeline(text-generation, modelQwen/Qwen2.5-1.5B) print(pipe(做一块好蛋糕的秘诀是))跑一下看看首次运行会下载模型权重并缓存到本机之后秒开。返回是一个列表generated_text字段里就是补全后的完整句子。到这里你已经有了可复用的推理管道。进阶给图片做目标检测再扩到批量推理 目标检测是验证 pipeline 最直观的任务输入一张图输出哪里有什么。from transformers import pipeline detector pipeline(object-detection, modelfacebook/detr-resnet-50) result detector(tests/fixtures/tests_samples/COCO/apple.jpg, top_k5) print(result[0][label], result[0][score]) # 类别名与置信度这就是仓库自带的测试样本。DETR 会返回 apple 类别、置信度分数和 x1、y1、x2、y2 四个包围盒坐标。想从单张走向批量只需掌握三个参数top_k控制每张图最多返回几个框画面越杂可以调大score是置信度业务里通常先按阈值过滤再落库想换检测器只改model参数其余代码一字不动。完整工作流从图片文件夹到一份标注清单 批量推理就是把单次调用放进循环。假设photos目录里放着一批厨房照片这类场景图里人和披萨都会进框正好检验top_k与置信度过滤的效果。流程只有三步扫描文件夹、逐张检测、结果写入 JSON。from transformers import pipeline import json, os detector pipeline(object-detection, modelfacebook/detr-resnet-50) records [] for name in sorted(os.listdir(photos)): if not name.endswith((.jpg, .png)): continue for item in detector(fphotos/{name}, top_k5): records.append({image: name, **item}) json.dump(records, open(index.json, w), ensure_asciiFalse, indent2)跑完后index.json就是一份可检索的图像索引哪张图、什么物体、置信度多少、框在哪个位置。想再进一步把自有标注数据交给 Trainer微调入口在 src/transformers/trainer.py训练流程和推理共用同一套模型定义。源码导读transformers 核心实现都藏在哪src/transformers/pipelines/全部任务 pipeline高层 API 的落点每种任务一个文件src/transformers/models/detr/DETR 的配置文件、图像预处理与模型定义一个模型一个目录src/transformers/configuration_utils.py配置解析逻辑模型结构参数都从这里进。想看懂某个模型怎么定义直接进models/下对应目录结构都是configuration image_processing modeling三件套。新手最常踩的几个坑 ⚠️首次运行下载慢模型权重从 Hub 拉取并缓存本机网速决定第一印象。建议把 HF_HOME 指到空间充裕的磁盘常用权重提前下载。显存不够大模型默认精度比较吃显存。先换小一号的模型验证流程跑通后再考虑量化加载。任务名大小写敏感pipeline 的 task 参数写错会直接抛错。对照官方任务列表逐个核对写法。图片尺寸不用操心AutoImageProcessor 统一负责缩放与归一化传文件路径或 PIL 对象都可以。CPU 上慢属正常检测模型在 CPU 上每张可能要几秒。生产环境建议上 GPU推理时改用 float16。模型定义、推理、训练三件事现在都在同一个包里了。装上它从一张图片的检测开始给你的第一个应用跑起来 【免费下载链接】transformers Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.项目地址: https://gitcode.com/GitHub_Trending/tra/transformers创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考