YOLOv5获得大中小目标的AP和AR指标(自制数据集)

文章目录

  • 前言
  • 一、运行示例
  • 二、参考
  • 三、方法
    • 1.安装pycocotools库
    • 2.YOLOv5代码修改
    • 3.制作.json文件
    • 4.运行程序
  • 附录
  • 总结


前言

本文简要介绍YOLOv5如何调用pycocotools得到大中小目标的AP和AR指标,评价自制数据集。

  • 代码版本-----YOLOv5_6.0版本。
  • 数据集----Seaships7000数据集,共包含6类7000张船舶图片,其中测试集1400张。
  • 模型-----自制模型。

一、运行示例

话不多说,运行示例:

(pytorch1.8) zmy@

你可能感兴趣的:(python)