SSTato / YOLOv7_obb

Rotating object detection using YOLOv7 and CSL.
GNU General Public License v3.0
26 stars 3 forks source link

YOLOv7_obb

Rotating object detection using YOLOv7 and CSL.

项目用途

使用YOLOv7和CSL实现旋转目标检测,在DOTA数据集和HRSC2016数据集上均取得良好结果。 (train_ota.py已上传)

效果展示

DOTA数据集HBB(0.791) (click to expand)

DOTA数据集OBB(val) (0.7835) (click to expand)

mAP:78.35

class aps: [90.56903172 80.70941623 46.04763298 59.882317 80.20104153 87.27336943 89.98862253 90.8265387 84.66325542 89.90289903 76.64386534 74.47407929 76.2255546 67.98867669 79.85177155]

DOTA数据集OBB(test) (0.7840) (click to expand)

mAP:78.39785536928188

结果链接:https://drive.google.com/file/d/1i8mkKTJhfpilW6XEC02rPGTx_IOfo5pI/view?usp=sharing

HRSC2016数据集HBB(0.942) (click to expand)

HRSC2016数据集OBB(0.959) (click to expand)

文件下载

模型文件可在谷歌网盘中下载 (click to expand)

链接:https://drive.google.com/file/d/1bodl7IoA1xaqP-b9RXDvRUv6oSxNySP3/view?usp=sharing

链接:https://drive.google.com/file/d/1SLrafWiaoYzgjq9KlXuKr_-wn7-5Zbj2/view?usp=sharing