模型介绍
中文整理这里展示了一个代码片段,展示了如何使用印章检测模型:
from ultralytics import YOLO
import os
from pathlib import Path
- 加载训练好的模型
model = YOLO('best.pt')
- 设置输入和输出文件夹路径
input folder = 'photo seal'
output folder = 'results seal'
创建输出目录
Path(output folder).mkdir(exist ok=True)
- 支持的图片格式
image extensions = ('.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.JPG', '.JPEG', '.PNG', '.BMP', '.TIFF')
- 获取所有图片文件
image files = [f for f in os.listdir(input folder) if f.lower().endswith(image extensions)]
- 批量推理并保存结果
for filename in image files:
img path = os.path.join(input folder, filename)
results = model(img path)
output path = os.path.join(output folder, f"result {filename}")
results[0].save(filename=output path)
SDK下载
Git下载
如果您是本模型的贡献者,我们邀请您根据模型贡献文档,及时完善模型卡片内容。
优先采用原作者提供的中文模型卡片;仅有英文说明时由闲社AI翻译整理。重要参数、许可证和商用范围请以原文为准。
你将获得什么
完整仓库方式包含该固定版本中的权重、配置、分词器及说明文件;指定版本方式只下载所选文件或整组分片。多模态模型的投影文件、配置等依赖,请按作者说明配齐。
完整仓库可能包含多个权重格式,因此需要的磁盘空间可能大于单一格式的模型大小。下载文件不等于完成模型部署。
使用前须知
运行模型需要的设备、工具与依赖,以原作者说明为准。
是否允许商用、修改和分发,请查看模型许可证。
闲社提供资料与镜像下载方法,不代表已运行模型或完成安全审计。下载的代码应先检查,再执行。
资料或下载有问题?
登录后可提交反馈。
