├── Arial.ttf ├── CONTRIBUTING.md ├── DOTA_devkit ├── DOTA.py ├── DOTA2COCO.py ├── DOTA2JSON.py ├── ImgSplit.py ├── ImgSplit_multi_process.py ├── ResultEnsembleNMS_multi_process.py ├── ResultMerge.py ├── ResultMerge_multi_process.py ├── SplitOnlyImage.py ├── SplitOnlyImage_multi_process.py ├── __init__.py ├── dota_evaluation_task1.py ├── dota_evaluation_task2.py ├── dota_poly2rbox.py ├── dota_utils.py ├── hrsc2016_evaluation.py ├── mAOE_evaluation.py ├── poly_nms_gpu │ ├── Makefile │ ├── __init__.py │ ├── nms_wrapper.py │ ├── poly_nms.cpp │ ├── poly_nms.hpp │ ├── poly_nms.pyx │ ├── poly_nms_kernel.cu │ ├── poly_nms_test.py │ ├── poly_overlaps.cpp │ ├── poly_overlaps.hpp │ ├── poly_overlaps.pyx │ ├── poly_overlaps_kernel.cu │ └── setup.py ├── polyiou.cpp ├── polyiou.h ├── polyiou.i ├── polyiou.py ├── polyiou_wrap.cxx ├── prepare_dota1_ms.py ├── prepare_hrsc2016.py ├── results_ensemble.py ├── results_obb2hbb.py ├── setup.py └── ucasaod_evaluation.py ├── Dockerfile ├── LICENSE ├── README.md ├── data ├── DroneVehicle_poly.yaml ├── dotav15_poly.yaml ├── dotav1_poly.yaml ├── hyps │ └── obb │ │ ├── hyp.finetune_DroneVehicle.yaml │ │ ├── hyp.finetune_dota.yaml │ │ ├── hyp.finetune_dota_CloseAug.yaml │ │ └── hyp.paper.yaml ├── scripts │ └── download_weights.sh ├── yolov5obb_demo.yaml └── yolov5obb_demo_split.yaml ├── dataset └── dataset_demo │ ├── images │ └── P0032.png │ ├── imgnamefile.txt │ └── labelTxt │ └── P0032.txt ├── detect.py ├── docs ├── ChangeLog.md ├── GetStart.md ├── YOLOv5_README.md ├── detection.png ├── install.md ├── results.png └── train_batch6.jpg ├── export.py ├── hubconf.py ├── models ├── __init__.py ├── common.py ├── experimental.py ├── hub │ ├── anchors.yaml │ ├── yolov3-spp.yaml │ ├── yolov3-tiny.yaml │ ├── yolov3.yaml │ ├── yolov5-bifpn.yaml │ ├── yolov5-fpn.yaml │ ├── yolov5-p2.yaml │ ├── yolov5-p6.yaml │ ├── yolov5-p7.yaml │ ├── yolov5-panet.yaml │ ├── yolov5l6.yaml │ ├── yolov5m6.yaml │ ├── yolov5n6.yaml │ ├── yolov5s-ghost.yaml │ ├── yolov5s-transformer.yaml │ ├── yolov5s6.yaml │ └── yolov5x6.yaml ├── tf.py ├── yolo.py ├── yolov5l.yaml ├── yolov5m.yaml ├── yolov5n.yaml ├── yolov5s.yaml └── yolov5x.yaml ├── requirements.txt ├── setup.cfg ├── sh └── ddp_train.sh ├── test.txt ├── tools ├── TestJson2VocClassTxt.py └── Xml2Txt.py ├── train.py ├── tutorial.ipynb ├── utils ├── __init__.py ├── activations.py ├── augmentations.py ├── autoanchor.py ├── autobatch.py ├── aws │ ├── __init__.py │ ├── mime.sh │ ├── resume.py │ └── userdata.sh ├── callbacks.py ├── datasets.py ├── downloads.py ├── flask_rest_api │ ├── README.md │ ├── example_request.py │ └── restapi.py ├── general.py ├── google_app_engine │ ├── Dockerfile │ ├── additional_requirements.txt │ └── app.yaml ├── loggers │ ├── __init__.py │ └── wandb │ │ ├── README.md │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ └── wandb_utils.cpython-39.pyc │ │ ├── log_dataset.py │ │ ├── sweep.py │ │ ├── sweep.yaml │ │ └── wandb_utils.py ├── loss.py ├── metrics.py ├── nms_rotated │ ├── __init__.py │ ├── nms_rotated_wrapper.py │ ├── setup.py │ └── src │ │ ├── box_iou_rotated_utils.h │ │ ├── nms_rotated_cpu.cpp │ │ ├── nms_rotated_cuda.cu │ │ ├── nms_rotated_ext.cpp │ │ ├── poly_nms_cpu.cpp │ │ └── poly_nms_cuda.cu ├── plots.py ├── rboxs_utils.py └── torch_utils.py └── val.py /Arial.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/Arial.ttf -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /DOTA_devkit/DOTA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/DOTA.py -------------------------------------------------------------------------------- /DOTA_devkit/DOTA2COCO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/DOTA2COCO.py -------------------------------------------------------------------------------- /DOTA_devkit/DOTA2JSON.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/DOTA2JSON.py -------------------------------------------------------------------------------- /DOTA_devkit/ImgSplit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/ImgSplit.py -------------------------------------------------------------------------------- /DOTA_devkit/ImgSplit_multi_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/ImgSplit_multi_process.py -------------------------------------------------------------------------------- /DOTA_devkit/ResultEnsembleNMS_multi_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/ResultEnsembleNMS_multi_process.py -------------------------------------------------------------------------------- /DOTA_devkit/ResultMerge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/ResultMerge.py -------------------------------------------------------------------------------- /DOTA_devkit/ResultMerge_multi_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/ResultMerge_multi_process.py -------------------------------------------------------------------------------- /DOTA_devkit/SplitOnlyImage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/SplitOnlyImage.py -------------------------------------------------------------------------------- /DOTA_devkit/SplitOnlyImage_multi_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/SplitOnlyImage_multi_process.py -------------------------------------------------------------------------------- /DOTA_devkit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DOTA_devkit/dota_evaluation_task1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/dota_evaluation_task1.py -------------------------------------------------------------------------------- /DOTA_devkit/dota_evaluation_task2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/dota_evaluation_task2.py -------------------------------------------------------------------------------- /DOTA_devkit/dota_poly2rbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/dota_poly2rbox.py -------------------------------------------------------------------------------- /DOTA_devkit/dota_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/dota_utils.py -------------------------------------------------------------------------------- /DOTA_devkit/hrsc2016_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/hrsc2016_evaluation.py -------------------------------------------------------------------------------- /DOTA_devkit/mAOE_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/mAOE_evaluation.py -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/Makefile -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/__init__.py -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/nms_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/nms_wrapper.py -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/poly_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/poly_nms.cpp -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/poly_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/poly_nms.hpp -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/poly_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/poly_nms.pyx -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/poly_nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/poly_nms_kernel.cu -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/poly_nms_test.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/poly_overlaps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/poly_overlaps.cpp -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/poly_overlaps.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/poly_overlaps.hpp -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/poly_overlaps.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/poly_overlaps.pyx -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/poly_overlaps_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/poly_overlaps_kernel.cu -------------------------------------------------------------------------------- /DOTA_devkit/poly_nms_gpu/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/poly_nms_gpu/setup.py -------------------------------------------------------------------------------- /DOTA_devkit/polyiou.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/polyiou.cpp -------------------------------------------------------------------------------- /DOTA_devkit/polyiou.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/polyiou.h -------------------------------------------------------------------------------- /DOTA_devkit/polyiou.i: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/polyiou.i -------------------------------------------------------------------------------- /DOTA_devkit/polyiou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/polyiou.py -------------------------------------------------------------------------------- /DOTA_devkit/polyiou_wrap.cxx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/polyiou_wrap.cxx -------------------------------------------------------------------------------- /DOTA_devkit/prepare_dota1_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/prepare_dota1_ms.py -------------------------------------------------------------------------------- /DOTA_devkit/prepare_hrsc2016.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/prepare_hrsc2016.py -------------------------------------------------------------------------------- /DOTA_devkit/results_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/results_ensemble.py -------------------------------------------------------------------------------- /DOTA_devkit/results_obb2hbb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/results_obb2hbb.py -------------------------------------------------------------------------------- /DOTA_devkit/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/setup.py -------------------------------------------------------------------------------- /DOTA_devkit/ucasaod_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/DOTA_devkit/ucasaod_evaluation.py -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/README.md -------------------------------------------------------------------------------- /data/DroneVehicle_poly.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/DroneVehicle_poly.yaml -------------------------------------------------------------------------------- /data/dotav15_poly.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/dotav15_poly.yaml -------------------------------------------------------------------------------- /data/dotav1_poly.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/dotav1_poly.yaml -------------------------------------------------------------------------------- /data/hyps/obb/hyp.finetune_DroneVehicle.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/hyps/obb/hyp.finetune_DroneVehicle.yaml -------------------------------------------------------------------------------- /data/hyps/obb/hyp.finetune_dota.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/hyps/obb/hyp.finetune_dota.yaml -------------------------------------------------------------------------------- /data/hyps/obb/hyp.finetune_dota_CloseAug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/hyps/obb/hyp.finetune_dota_CloseAug.yaml -------------------------------------------------------------------------------- /data/hyps/obb/hyp.paper.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/hyps/obb/hyp.paper.yaml -------------------------------------------------------------------------------- /data/scripts/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/scripts/download_weights.sh -------------------------------------------------------------------------------- /data/yolov5obb_demo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/yolov5obb_demo.yaml -------------------------------------------------------------------------------- /data/yolov5obb_demo_split.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/data/yolov5obb_demo_split.yaml -------------------------------------------------------------------------------- /dataset/dataset_demo/images/P0032.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/dataset/dataset_demo/images/P0032.png -------------------------------------------------------------------------------- /dataset/dataset_demo/imgnamefile.txt: -------------------------------------------------------------------------------- 1 | P0032 2 | -------------------------------------------------------------------------------- /dataset/dataset_demo/labelTxt/P0032.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/dataset/dataset_demo/labelTxt/P0032.txt -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/detect.py -------------------------------------------------------------------------------- /docs/ChangeLog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/docs/ChangeLog.md -------------------------------------------------------------------------------- /docs/GetStart.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/docs/GetStart.md -------------------------------------------------------------------------------- /docs/YOLOv5_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/docs/YOLOv5_README.md -------------------------------------------------------------------------------- /docs/detection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/docs/detection.png -------------------------------------------------------------------------------- /docs/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/docs/install.md -------------------------------------------------------------------------------- /docs/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/docs/results.png -------------------------------------------------------------------------------- /docs/train_batch6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/docs/train_batch6.jpg -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/export.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/hubconf.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/hub/anchors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/anchors.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov3-tiny.yaml -------------------------------------------------------------------------------- /models/hub/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov3.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-bifpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5-bifpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5-p2.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5-p6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5-p7.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /models/hub/yolov5l6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5l6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5m6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5m6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5n6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5n6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s-ghost.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5s-ghost.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s-transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5s-transformer.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5s6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5x6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/hub/yolov5x6.yaml -------------------------------------------------------------------------------- /models/tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/tf.py -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/yolov5l.yaml -------------------------------------------------------------------------------- /models/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/yolov5m.yaml -------------------------------------------------------------------------------- /models/yolov5n.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/yolov5n.yaml -------------------------------------------------------------------------------- /models/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/yolov5s.yaml -------------------------------------------------------------------------------- /models/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/models/yolov5x.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/setup.cfg -------------------------------------------------------------------------------- /sh/ddp_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/sh/ddp_train.sh -------------------------------------------------------------------------------- /test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/test.txt -------------------------------------------------------------------------------- /tools/TestJson2VocClassTxt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/tools/TestJson2VocClassTxt.py -------------------------------------------------------------------------------- /tools/Xml2Txt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/tools/Xml2Txt.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/train.py -------------------------------------------------------------------------------- /tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/tutorial.ipynb -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/autobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/autobatch.py -------------------------------------------------------------------------------- /utils/aws/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/aws/mime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/aws/mime.sh -------------------------------------------------------------------------------- /utils/aws/resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/aws/resume.py -------------------------------------------------------------------------------- /utils/aws/userdata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/aws/userdata.sh -------------------------------------------------------------------------------- /utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/callbacks.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/downloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/downloads.py -------------------------------------------------------------------------------- /utils/flask_rest_api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/flask_rest_api/README.md -------------------------------------------------------------------------------- /utils/flask_rest_api/example_request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/flask_rest_api/example_request.py -------------------------------------------------------------------------------- /utils/flask_rest_api/restapi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/flask_rest_api/restapi.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_app_engine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/google_app_engine/Dockerfile -------------------------------------------------------------------------------- /utils/google_app_engine/additional_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/google_app_engine/additional_requirements.txt -------------------------------------------------------------------------------- /utils/google_app_engine/app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/google_app_engine/app.yaml -------------------------------------------------------------------------------- /utils/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/loggers/__init__.py -------------------------------------------------------------------------------- /utils/loggers/wandb/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/loggers/wandb/README.md -------------------------------------------------------------------------------- /utils/loggers/wandb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/loggers/wandb/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/loggers/wandb/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /utils/loggers/wandb/__pycache__/wandb_utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/loggers/wandb/__pycache__/wandb_utils.cpython-39.pyc -------------------------------------------------------------------------------- /utils/loggers/wandb/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/loggers/wandb/log_dataset.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/loggers/wandb/sweep.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/loggers/wandb/sweep.yaml -------------------------------------------------------------------------------- /utils/loggers/wandb/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/loggers/wandb/wandb_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/nms_rotated/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/nms_rotated/__init__.py -------------------------------------------------------------------------------- /utils/nms_rotated/nms_rotated_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/nms_rotated/nms_rotated_wrapper.py -------------------------------------------------------------------------------- /utils/nms_rotated/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/nms_rotated/setup.py -------------------------------------------------------------------------------- /utils/nms_rotated/src/box_iou_rotated_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/nms_rotated/src/box_iou_rotated_utils.h -------------------------------------------------------------------------------- /utils/nms_rotated/src/nms_rotated_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/nms_rotated/src/nms_rotated_cpu.cpp -------------------------------------------------------------------------------- /utils/nms_rotated/src/nms_rotated_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/nms_rotated/src/nms_rotated_cuda.cu -------------------------------------------------------------------------------- /utils/nms_rotated/src/nms_rotated_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/nms_rotated/src/nms_rotated_ext.cpp -------------------------------------------------------------------------------- /utils/nms_rotated/src/poly_nms_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/nms_rotated/src/poly_nms_cpu.cpp -------------------------------------------------------------------------------- /utils/nms_rotated/src/poly_nms_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/nms_rotated/src/poly_nms_cuda.cu -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/rboxs_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/rboxs_utils.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hukaixuan19970627/yolov5_obb/HEAD/val.py --------------------------------------------------------------------------------