YOLOv10 - Berkeley DeepDrive (BDD) 100K Vanilla
YOLOv10 model trained from scratch on Berkeley DeepDrive (BDD) 100K dataset for object detection in autonomous driving scenarios.
Model Details
- Model Type: YOLOv10 Object Detection
- Dataset: Berkeley DeepDrive (BDD) 100K
- Training Method: trained from scratch
- Framework: PyTorch/Ultralytics
- Task: Object Detection
Dataset Information
This model was trained on the Berkeley DeepDrive (BDD) 100K dataset, which contains the following object classes:
car, truck, bus, motorcycle, bicycle, person, traffic light, traffic sign, train, rider
Dataset-specific Details:
Berkeley DeepDrive (BDD) 100K Dataset:
- 100,000+ driving images with diverse weather and lighting conditions
- Designed for autonomous driving applications
- Contains urban driving scenarios from multiple cities
- Annotations include bounding boxes for vehicles, pedestrians, and traffic elements
Usage
This model can be used with the Ultralytics YOLOv10 framework:
1from ultralytics import YOLO
2
3# Load the model
4model = YOLO('path/to/best.pt')
5
6# Run inference
7results = model('path/to/image.jpg')
8
9# Process results
10for result in results:
11 boxes = result.boxes.xyxy # bounding boxes
12 scores = result.boxes.conf # confidence scores
13 classes = result.boxes.cls # class predictions
Model Performance
This model was trained from scratch on the Berkeley DeepDrive (BDD) 100K dataset using YOLOv10 architecture.
Intended Use
- Primary Use: Object detection in autonomous driving scenarios
- Suitable for: Research, development, and deployment of object detection systems
- Limitations: Performance may vary on images significantly different from the training distribution
Citation
If you use this model, please cite:
1@article{yolov10,
2 title={YOLOv10: Real-Time End-to-End Object Detection},
3 author={Wang, Ao and Chen, Hui and Liu, Lihao and Chen, Kai and Lin, Zijia and Han, Jungong and Ding, Guiguang},
4 journal={arXiv preprint arXiv:2405.14458},
5 year={2024}
6}
License
This model is released under the MIT License.
Keywords
YOLOv10, Object Detection, Computer Vision, BDD 100K, Autonomous Driving, Deep Learning