Polygon-YOLOv5自定义数据集使用指南.pdfVIP

步骤1.

安装必要软件包及CUDA扩展

请按以下步骤安装所需依赖及用于计算多边形框IOU（交并比）的CUDA扩展。

CUDA扩展安装指南（命令行）

#

切换到polygon‑yolov5项目目录

cd

polygon‑yolov5

#

安装Python依赖包

#

如需Albumentation数据增强功能，请检查requirements.txt

pip

install

‑r

requirements.txt

#

安装CUDA扩展

_

cd

utils/ioucuda

python

setup.py

install

cd

..

cd

..

步骤2.

数据标注与训练‑验证‑测试集划分

_

本步骤旨在获取多边形标注数据（classid,

x1,

y1,

x2,

y2,

x3,

y3,

x4,

y4）用于训练、

验证及最终测试。您可使用任何工具或应用程序进行标注，但需注意：确保四个角点按顺序