1.文件目录

• test_onnx.py:预测的py文件

• img:分类的原始图片

• model: onnx模型

2.安装的库

• import numpy as np
• import onnxruntime as rt
• from PIL import Image
• from PIL import ImageDraw, ImageFont
• import colorsys

3.使用说明

模型文件链接:

• https://pan.baidu.com/s/1E8THOfmvD1H6US_j_B2gtA

• 提取码:4q77

  模型放在model目录下

• python3 test_onnx.py

4.检测结果

• DETR模型介绍