Fine-tuned YOLOv11x object detector for aerial imagery using the VisDrone benchmark dataset.
Metrics reported in this model card are computed on the VisDrone test set with ground-truth annotations available for evaluation.
VisDrone is a large-scale benchmark for object detection in aerial imagery captured from unmanned aerial vehicles (UAVs).
1from huggingface_hub import hf_hub_download
2from ultralytics import YOLO
3
4weights = hf_hub_download(
5 repo_id="dronefreak/yolov11x-visdrone",
6 filename="best.pt"
7)
8
9model = YOLO(weights)
1results = model.predict(
2 source="image.jpg",
3 conf=0.25
4)
5
6results[0].show()
1best.pt
2results.csv
3args.yaml
4BoxPR_curve.png
5BoxF1_curve.png
6confusion_matrix.png
7assets/visdrone_showcase.gif
8README.md
These models were trained using the VisDrone Dataset Python Toolkit, an open-source framework for aerial object detection research and benchmarking on the VisDrone dataset.
If you find these models useful, please consider starring the repository.
1@article{visdrone2019,
2 title={Vision Meets Drones: A Challenge},
3 author={Zhu, Pengfei and Wen, Longyin and Du, Dawei and Bian, Xiao and Ling, Haibin and Hu, Qinghua},
4 journal={International Journal of Computer Vision},
5 year={2021}
6}
7
8@software{Saksena_VisDrone_Detection_Toolkit_2025,
9 author = {Saksena, Saumya Kumaar},
10 title = {VisDrone Detection Toolkit: Modern PyTorch Implementation for Aerial Object Detection},
11 url = {https://github.com/dronefreak/VisDrone-dataset-python-toolkit},
12 version = {2.0.0},
13 year = {2025}
14}