Data

Tools

Indexes

Applications

Experiments

Open Source Science

https://github.com/capjamesg/yolov5_obb

yolov5 + csl_label.(Oriented Object Detection)（Rotation Detection）（Rotated BBox）基于yolov5的旋转目标检测

https://github.com/capjamesg/yolov5_obb

Science Score: 10.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • ○
    CITATION.cff file
  • ○
    codemeta.json file
  • ○
    .zenodo.json file
  • ○
    DOI references
  • ✓
    Academic publication links
    Links to: arxiv.org
  • ○
    Academic email domains
  • ○
    Institutional organization owner
  • ○
    JOSS paper metadata
  • ○
    Scientific vocabulary similarity
    Low similarity (8.8%) to scientific vocabulary
Last synced: 9 months ago · JSON representation

Repository

yolov5 + csl_label.(Oriented Object Detection)（Rotation Detection）（Rotated BBox）基于yolov5的旋转目标检测

Basic Info
  • Host: GitHub
  • Owner: capjamesg
  • License: gpl-3.0
  • Default Branch: master
  • Homepage:
  • Size: 17.6 MB
Statistics
  • Stars: 0
  • Watchers: 1
  • Forks: 0
  • Open Issues: 1
  • Releases: 0
Fork of hukaixuan19970627/yolov5_obb
Created over 2 years ago · Last pushed over 1 year ago
Metadata Files
Readme Contributing License

README.md

Yolov5 for Oriented Object Detection

图片 train_batch0.jpg results.png

The code for the implementation of “Yolov5 + Circular Smooth Label”.

Results and Models

The results on DOTAsubsize1024gap200_rate1.0 test-dev set are shown in the table below. (password: yolo)

|Model
(download link) |Size
(pixels) | TTA
(multi-scale/
rotate testing) | OBB mAPtest
0.5
DOTAv1.0 | OBB mAPtest
0.5
DOTAv1.5 | OBB mAPtest
0.5
DOTAv2.0 | Speed
CPU b1
(ms)|Speed
2080Ti b1
(ms) |Speed
2080Ti b16
(ms) |params
(M) |FLOPs
@640 (B) | ---- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- |yolov5m [baidu/google] |1024 | × |77.3 |73.2 |58.0 |328.2 |16.9 |11.3 |21.6 |50.5
|yolov5s [baidu] |1024 | × |76.8 |- |- |- |15.6 | - |7.5 |17.5
|yolov5n [baidu] |1024 | × |73.3 |- |- |- |15.2 | - |2.0 |5.0

Table Notes (click to expand / **点我看更多**) * All checkpoints are trained to 300 epochs with [COCO pre-trained checkpoints](https://github.com/ultralytics/yolov5/releases/tag/v6.0), default settings and hyperparameters. * **mAPtest dota** values are for single-model single-scale on [DOTA](https://captain-whu.github.io/DOTA/index.html)(1024,1024,200,1.0) dataset.
Reproduce Example: ```shell python val.py --data 'data/dotav15_poly.yaml' --img 1024 --conf 0.01 --iou 0.4 --task 'test' --batch 16 --save-json --name 'dotav15_test_split' python tools/TestJson2VocClassTxt.py --json_path 'runs/val/dotav15_test_split/best_obb_predictions.json' --save_path 'runs/val/dotav15_test_split/obb_predictions_Txt' python DOTA_devkit/ResultMerge_multi_process.py --scrpath 'runs/val/dotav15_test_split/obb_predictions_Txt' --dstpath 'runs/val/dotav15_test_split/obb_predictions_Txt_Merged' zip the poly format results files and submit it to https://captain-whu.github.io/DOTA/evaluation.html ``` * **Speed** averaged over DOTAv1.5 val_split_subsize1024_gap200 images using a 2080Ti gpu. NMS + pre-process times is included.
Reproduce by `python val.py --data 'data/dotav15_poly.yaml' --img 1024 --task speed --batch 1`

Updates

  • [2022/1/7] : Faster and stronger, some bugs fixed, yolov5 base version updated.

Installation

Please refer to install.md for installation and dataset preparation.

Getting Started

This repo is based on yolov5.

And this repo has been rebuilt, Please see GetStart.md for the Oriented Detection latest basic usage.

Acknowledgements

I have used utility functions from other wonderful open-source projects. Espeicially thank the authors of:

有问题反馈

在使用中有任何问题，建议先按照install.md检查环境依赖项，再按照GetStart.md检查使用流程是否正确，善用搜索引擎和github中的issue搜索框，可以极大程度上节省你的时间。

若遇到的是新问题，可以用以下联系方式跟我交流，为了提高沟通效率，请尽可能地提供相关信息以便我复现该问题。

  • 知乎（@略略略）
  • 代码问题提issues,其他问题请知乎上联系

关于作者

```javascript Name : "胡凯旋" describe myself："咸鱼一枚"

Owner

  • Name: James
  • Login: capjamesg
  • Kind: user
  • Location: Scotland
  • Company: @Roboflow

from words, wonder.

GitHub Events

Total
  • Pull request event: 1
  • Create event: 1
Last Year
  • Pull request event: 1
  • Create event: 1

Dependencies

Dockerfile docker
  • nvcr.io/nvidia/pytorch 21.10-py3 build
utils/google_app_engine/Dockerfile docker
  • gcr.io/google-appengine/python latest build
DOTA_devkit/poly_nms_gpu/setup.py pypi
DOTA_devkit/setup.py pypi
requirements.txt pypi
  • Pillow >=7.1.2
  • PyYAML >=5.3.1
  • matplotlib >=3.2.2
  • numpy >=1.18.5
  • opencv-python >=4.5.4
  • pandas >=1.1.4
  • requests >=2.23.0
  • scipy >=1.4.1
  • seaborn >=0.11.0
  • tensorboard >=2.4.1
  • thop *
  • torch >=1.7.0
  • torchvision >=0.8.1
  • tqdm >=4.41.0
utils/google_app_engine/additional_requirements.txt pypi
  • Flask ==1.0.2
  • gunicorn ==19.9.0
  • pip ==21.1
utils/nms_rotated/setup.py pypi