这里展示了一个代码片段，展示了如何使用商标检测模型： from ultralytics import YOLO import os from pathlib import Path

1. 加载训练好的模型

model = YOLO('goods_detect.pt')

2. 设置输入和输出文件夹路径

input_folder = 'photo' output_folder = 'results_goods'

创建输出目录

Path(output_folder).mkdir(exist_ok=True)

3. 支持的图片格式

image_extensions = ('.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.JPG', '.JPEG', '.PNG', '.BMP', '.TIFF')

4. 获取所有图片文件

image_files = [f for f in os.listdir(input_folder) if f.lower().endswith(image_extensions)]

5. 批量推理并保存结果

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下载

#安装ModelScope
pip install modelscope

#SDK模型下载
from modelscope import snapshot_download
model_dir = snapshot_download('SdtDevAi/trademark_detect')

Git下载

#Git模型下载
git clone https://www.modelscope.cn/SdtDevAi/trademark_detect.git

如果您是本模型的贡献者，我们邀请您根据模型贡献文档，及时完善模型卡片内容。