├── .gitignore ├── Readme.md ├── data ├── DOTA.yaml ├── UCAS_AOD.yaml └── hyp.yaml ├── datasets ├── DOTA_dataset.py ├── UCASAOD_dataset.py ├── base_dataset.py └── custom_dataset.py ├── detect.py ├── display_inputs.py ├── docker └── Dockerfile ├── images ├── angle.png └── loss.png ├── lib ├── __init__.py ├── augmentations.py ├── general.py ├── load.py ├── logger.py ├── loss.py └── plot.py ├── model ├── __init__.py ├── backbone.py ├── neck.py ├── utils.py ├── yolo.py └── yololayer.py ├── outputs ├── DOTA │ ├── P0006.png │ ├── P0024.png │ ├── P0031.png │ └── P0197.png ├── UCAS_AOD │ ├── P0292.png │ └── P0769.png └── trash │ ├── 480.jpg │ └── 540.jpg ├── requirements.txt ├── scripts ├── detect.sh ├── test.sh └── train.sh ├── test.py ├── train.py ├── weights └── .gitkeep └── xml2txt.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/.gitignore -------------------------------------------------------------------------------- /Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/Readme.md -------------------------------------------------------------------------------- /data/DOTA.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/data/DOTA.yaml -------------------------------------------------------------------------------- /data/UCAS_AOD.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/data/UCAS_AOD.yaml -------------------------------------------------------------------------------- /data/hyp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/data/hyp.yaml -------------------------------------------------------------------------------- /datasets/DOTA_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/datasets/DOTA_dataset.py -------------------------------------------------------------------------------- /datasets/UCASAOD_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/datasets/UCASAOD_dataset.py -------------------------------------------------------------------------------- /datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/datasets/base_dataset.py -------------------------------------------------------------------------------- /datasets/custom_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/datasets/custom_dataset.py -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/detect.py -------------------------------------------------------------------------------- /display_inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/display_inputs.py -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /images/angle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/images/angle.png -------------------------------------------------------------------------------- /images/loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/images/loss.png -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/lib/augmentations.py -------------------------------------------------------------------------------- /lib/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/lib/general.py -------------------------------------------------------------------------------- /lib/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/lib/load.py -------------------------------------------------------------------------------- /lib/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/lib/logger.py -------------------------------------------------------------------------------- /lib/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/lib/loss.py -------------------------------------------------------------------------------- /lib/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/lib/plot.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/model/backbone.py -------------------------------------------------------------------------------- /model/neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/model/neck.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/model/utils.py -------------------------------------------------------------------------------- /model/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/model/yolo.py -------------------------------------------------------------------------------- /model/yololayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/model/yololayer.py -------------------------------------------------------------------------------- /outputs/DOTA/P0006.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/outputs/DOTA/P0006.png -------------------------------------------------------------------------------- /outputs/DOTA/P0024.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/outputs/DOTA/P0024.png -------------------------------------------------------------------------------- /outputs/DOTA/P0031.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/outputs/DOTA/P0031.png -------------------------------------------------------------------------------- /outputs/DOTA/P0197.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/outputs/DOTA/P0197.png -------------------------------------------------------------------------------- /outputs/UCAS_AOD/P0292.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/outputs/UCAS_AOD/P0292.png -------------------------------------------------------------------------------- /outputs/UCAS_AOD/P0769.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/outputs/UCAS_AOD/P0769.png -------------------------------------------------------------------------------- /outputs/trash/480.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/outputs/trash/480.jpg -------------------------------------------------------------------------------- /outputs/trash/540.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/outputs/trash/540.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | opencv-python 2 | tensorboard 3 | tqdm 4 | pyyaml 5 | colorlog -------------------------------------------------------------------------------- /scripts/detect.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/scripts/detect.sh -------------------------------------------------------------------------------- /scripts/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/scripts/test.sh -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/train.py -------------------------------------------------------------------------------- /weights/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /xml2txt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingkunwu/R-YOLOv4/HEAD/xml2txt.py --------------------------------------------------------------------------------