UAV Vehicle Detection

UAV航拍车辆目标检测。

数据集

• 11850张可见光图像 (640x512)
• 11类目标：car, suv, van, bus, freight_car, truck, motorcycle, trailer, excavator, crane, tank_truck
• VOC XML标注格式，含旋转框

模型

YOLOv8m + 预训练权重

使用流程

# 1. 训练
python yolo_train.py

# 2. 推理
python yolo_predict.py

# 3. 生成提交文件
python convert_output.py

输出

提交文件夹 submission/ 包含11个类别txt文件，格式：

image_id confidence x1 y1 x2 y2 x3 y3 x4 y4

依赖

pip install -r requirements.txt