├── .gitignore ├── LICENSE ├── README.md ├── VOCdevkit └── VOC2007 │ ├── Annotations │ └── README.md │ ├── ImageSets │ └── Main │ │ └── README.md │ └── JPEGImages │ └── README.md ├── convert_trt.py ├── get_map.py ├── hrsc_annotation.py ├── img └── test.jpg ├── kmeans_for_anchors.py ├── logs └── README.md ├── model_data ├── coco_classes.txt ├── simhei.ttf ├── uav_classes.txt ├── voc_classes.txt └── yolo_anchors.txt ├── nets ├── __init__.py ├── backbone.py ├── yolo.py └── yolo_training.py ├── predict.py ├── requirements.txt ├── summary.py ├── train.py ├── utils ├── __init__.py ├── callbacks.py ├── dataloader.py ├── kld_loss.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 ├── utils.py ├── utils_bbox.py ├── utils_fit.py ├── utils_map.py └── utils_rbox.py ├── utils_coco ├── coco_annotation.py └── get_map_coco.py ├── voc_annotation.py ├── yolo.py └── 常见问题汇总.md /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/README.md -------------------------------------------------------------------------------- /VOCdevkit/VOC2007/Annotations/README.md: -------------------------------------------------------------------------------- 1 | 存放标签文件 -------------------------------------------------------------------------------- /VOCdevkit/VOC2007/ImageSets/Main/README.md: -------------------------------------------------------------------------------- 1 | 存放训练索引文件 -------------------------------------------------------------------------------- /VOCdevkit/VOC2007/JPEGImages/README.md: -------------------------------------------------------------------------------- 1 | 存放图片文件 -------------------------------------------------------------------------------- /convert_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/convert_trt.py -------------------------------------------------------------------------------- /get_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/get_map.py -------------------------------------------------------------------------------- /hrsc_annotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/hrsc_annotation.py -------------------------------------------------------------------------------- /img/test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/img/test.jpg -------------------------------------------------------------------------------- /kmeans_for_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/kmeans_for_anchors.py -------------------------------------------------------------------------------- /logs/README.md: -------------------------------------------------------------------------------- 1 | 训练好的权重会保存在这里 2 | -------------------------------------------------------------------------------- /model_data/coco_classes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/model_data/coco_classes.txt -------------------------------------------------------------------------------- /model_data/simhei.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/model_data/simhei.ttf -------------------------------------------------------------------------------- /model_data/uav_classes.txt: -------------------------------------------------------------------------------- 1 | car -------------------------------------------------------------------------------- /model_data/voc_classes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/model_data/voc_classes.txt -------------------------------------------------------------------------------- /model_data/yolo_anchors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/model_data/yolo_anchors.txt -------------------------------------------------------------------------------- /nets/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /nets/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/nets/backbone.py -------------------------------------------------------------------------------- /nets/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/nets/yolo.py -------------------------------------------------------------------------------- /nets/yolo_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/nets/yolo_training.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/requirements.txt -------------------------------------------------------------------------------- /summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/summary.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/callbacks.py -------------------------------------------------------------------------------- /utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/dataloader.py -------------------------------------------------------------------------------- /utils/kld_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/kld_loss.py -------------------------------------------------------------------------------- /utils/nms_rotated/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/nms_rotated/__init__.py -------------------------------------------------------------------------------- /utils/nms_rotated/nms_rotated_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/nms_rotated/nms_rotated_wrapper.py -------------------------------------------------------------------------------- /utils/nms_rotated/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/nms_rotated/setup.py -------------------------------------------------------------------------------- /utils/nms_rotated/src/box_iou_rotated_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/nms_rotated/src/box_iou_rotated_utils.h -------------------------------------------------------------------------------- /utils/nms_rotated/src/nms_rotated_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/nms_rotated/src/nms_rotated_cpu.cpp -------------------------------------------------------------------------------- /utils/nms_rotated/src/nms_rotated_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/nms_rotated/src/nms_rotated_cuda.cu -------------------------------------------------------------------------------- /utils/nms_rotated/src/nms_rotated_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/nms_rotated/src/nms_rotated_ext.cpp -------------------------------------------------------------------------------- /utils/nms_rotated/src/poly_nms_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/nms_rotated/src/poly_nms_cpu.cpp -------------------------------------------------------------------------------- /utils/nms_rotated/src/poly_nms_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/nms_rotated/src/poly_nms_cuda.cu -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/utils_bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/utils_bbox.py -------------------------------------------------------------------------------- /utils/utils_fit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/utils_fit.py -------------------------------------------------------------------------------- /utils/utils_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/utils_map.py -------------------------------------------------------------------------------- /utils/utils_rbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils/utils_rbox.py -------------------------------------------------------------------------------- /utils_coco/coco_annotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils_coco/coco_annotation.py -------------------------------------------------------------------------------- /utils_coco/get_map_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/utils_coco/get_map_coco.py -------------------------------------------------------------------------------- /voc_annotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/voc_annotation.py -------------------------------------------------------------------------------- /yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/yolo.py -------------------------------------------------------------------------------- /常见问题汇总.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Egrt/yolov7-tiny-obb/HEAD/常见问题汇总.md --------------------------------------------------------------------------------