In this repository, we provide popular models for each task in the BDD100K dataset. For each task in the dataset, we make publicly available the model weights, evaluation results, predictions, visualizations, as well as scripts to performance evaluation and visualization. The goal is to provide a set of competitive baselines to facilitate research and provide a common benchmark for comparison.
The number of pre-trained models in this zoo is 3️⃣0️⃣0️⃣. You can include your models in this repo as well! See contribution instructions.
This repository currently supports the tasks listed below. For more information about each task, click on the task name. We plan to support all tasks in the BDD100K dataset eventually; see the roadmap for our plan and progress.
- Image Tagging
- Object Detection
- Instance Segmentation
- Semantic Segmentation
- Panoptic Segmentation
- Drivable Area
- Multiple Object Tracking (MOT)
- Multiple Object Tracking and Segmentation (MOTS)
- Pose Estimation
If you have any questions, please go to the BDD100K discussions.
- Pose estimation
- Panoptic segmentation
- Lane marking
Please refer to the dataset preparation instructions for how to prepare and use the BDD100K dataset with the models.
To cite the BDD100K dataset in your paper,
@InProceedings{bdd100k,
author = {Yu, Fisher and Chen, Haofeng and Wang, Xin and Xian, Wenqi and Chen,
Yingying and Liu, Fangchen and Madhavan, Vashisht and Darrell, Trevor},
title = {BDD100K: A Diverse Driving Dataset for Heterogeneous Multitask Learning},
booktitle = {IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2020}
}