Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
58 changes: 43 additions & 15 deletions .github/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,10 +51,53 @@ data/
python scripts/train.py --available-models # list all 33 models
```

### Pretrained VisDrone Models

Pretrained VisDrone checkpoints for all supported YOLO architectures are available through the Hugging Face collection:

<https://huggingface.co/collections/dronefreak/visdrone-detection-model-zoo>

The collection includes model cards, benchmark results, evaluation visualizations, and ready-to-use weights for YOLOv8, YOLOv9, YOLOv10, YOLO11, and YOLO26 model families.

| Family | Available Models |
| ------- | ---------------- |
| YOLOv8 | n, s, m, x |
| YOLOv9 | c, m, e |
| YOLOv10 | n, l, x |
| YOLO11 | n, l, x |
| YOLO26 | n, l, x |

Individual model repositories can be accessed directly from the Hugging Face collection page.

---

## Usage

### Pre-trained Models

```bash
pip install ultralytics huggingface_hub
```

```python
from huggingface_hub import hf_hub_download
from ultralytics import YOLO

weights = hf_hub_download(
repo_id="dronefreak/yolov8m-visdrone",
filename="best.pt"
)

model = YOLO(weights)

results = model.predict(
source="image.jpg",
conf=0.25
)

results[0].show()
```

### Train

```bash
Expand Down Expand Up @@ -169,21 +212,6 @@ model = get_model("fasterrcnn_resnet50", num_classes=12, pretrained=True)

---

## Performance

Faster R-CNN ResNet50, VisDrone2019-DET-val (200 epochs, RTX 4070 Super):

| Metric | Score |
| --------- | ------------------- |
| F1 | 66.7% |
| Precision | 71.0% |
| Recall | 62.9% |
| Speed | 18 FPS (55ms/image) |

YOLO v8n after 1 epoch (untrained baseline): mAP@0.5 = 0.119, mAP@0.5:0.95 = 0.062.

---

## Development

```bash
Expand Down
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -182,3 +182,4 @@ temp/
# Cache
.cache/
*.cache
node_modules
74 changes: 0 additions & 74 deletions README.md

This file was deleted.

Loading