# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

data = {
    "YOLO12": {
        "author": "Yunjie Tian, Qixiang Ye, David Doermann",
        "date": "2024-02-18",
        "arxiv": "https://arxiv.org/abs/2502.12524",
        "github": "https://github.com/sunsmarterjie/yolov12",
        "docs": "https://docs.ultralytics.com/models/yolo12/",
        "performance": {
            "n": {"size": 640, "map": 40.6, "cpu": "", "t4": 1.64, "params": 2.6, "flops": 6.5},
            "s": {"size": 640, "map": 48.0, "cpu": "", "t4": 2.61, "params": 9.3, "flops": 21.4},
            "m": {"size": 640, "map": 52.5, "cpu": "", "t4": 4.86, "params": 20.2, "flops": 67.5},
            "l": {"size": 640, "map": 53.7, "cpu": "", "t4": 6.77, "params": 26.4, "flops": 88.9},
            "x": {"size": 640, "map": 55.2, "cpu": "", "t4": 11.79, "params": 59.1, "flops": 199.0},
        },
    },
    "YOLO11": {
        "author": "Glenn Jocher and Jing Qiu",
        "date": "2024-09-27",
        "arxiv": None,
        "github": "https://github.com/ultralytics/ultralytics",
        "docs": "https://docs.ultralytics.com/models/yolo11/",
        "performance": {
            "n": {"size": 640, "map": 39.5, "cpu": 56.1, "t4": 1.5, "params": 2.6, "flops": 6.5},
            "s": {"size": 640, "map": 47.0, "cpu": 90.0, "t4": 2.5, "params": 9.4, "flops": 21.5},
            "m": {"size": 640, "map": 51.5, "cpu": 183.2, "t4": 4.7, "params": 20.1, "flops": 68.0},
            "l": {"size": 640, "map": 53.4, "cpu": 238.6, "t4": 6.2, "params": 25.3, "flops": 86.9},
            "x": {"size": 640, "map": 54.7, "cpu": 462.8, "t4": 11.3, "params": 56.9, "flops": 194.9},
        },
    },
    "YOLOv10": {
        "author": "Ao Wang, Hui Chen, Lihao Liu, et al.",
        "date": "2024-05-23",
        "arxiv": "https://arxiv.org/abs/2405.14458",
        "github": "https://github.com/THU-MIG/yolov10",
        "docs": "https://docs.ultralytics.com/models/yolov10/",
        "performance": {
            "n": {"size": 640, "map": 39.5, "cpu": "", "t4": 1.56, "params": 2.3, "flops": 6.7},
            "s": {"size": 640, "map": 46.7, "cpu": "", "t4": 2.66, "params": 7.2, "flops": 21.6},
            "m": {"size": 640, "map": 51.3, "cpu": "", "t4": 5.48, "params": 15.4, "flops": 59.1},
            "b": {"size": 640, "map": 52.7, "cpu": "", "t4": 6.54, "params": 24.4, "flops": 92.0},
            "l": {"size": 640, "map": 53.3, "cpu": "", "t4": 8.33, "params": 29.5, "flops": 120.3},
            "x": {"size": 640, "map": 54.4, "cpu": "", "t4": 12.2, "params": 56.9, "flops": 160.4},
        },
    },
    "YOLOv9": {
        "author": "Wang, Chien-Yao and Liao, Hong-Yuan Mark",
        "date": "2024-02-21",
        "arxiv": "https://arxiv.org/abs/2402.13616",
        "github": "https://github.com/WongKinYiu/yolov9",
        "docs": "https://docs.ultralytics.com/models/yolov9/",
        "performance": {
            "t": {"size": 640, "map": 38.3, "cpu": "", "t4": 2.3, "params": 2.0, "flops": 7.7},
            "s": {"size": 640, "map": 46.8, "cpu": "", "t4": 3.54, "params": 7.1, "flops": 26.4},
            "m": {"size": 640, "map": 51.4, "cpu": "", "t4": 6.43, "params": 20.0, "flops": 76.3},
            "c": {"size": 640, "map": 53.0, "cpu": "", "t4": 7.16, "params": 25.3, "flops": 102.1},
            "e": {"size": 640, "map": 55.6, "cpu": "", "t4": 16.77, "params": 57.3, "flops": 189.0},
        },
    },
    "YOLOv8": {
        "author": "Glenn Jocher and Ayush Chaurasia and Jing Qiu",
        "date": "2023-01-10",
        "arxiv": None,
        "github": "https://github.com/ultralytics/ultralytics",
        "docs": "https://docs.ultralytics.com/models/yolov8/",
        "performance": {
            "n": {"size": 640, "map": 37.3, "cpu": 80.4, "t4": 1.47, "params": 3.2, "flops": 8.7},
            "s": {"size": 640, "map": 44.9, "cpu": 128.4, "t4": 2.66, "params": 11.2, "flops": 28.6},
            "m": {"size": 640, "map": 50.2, "cpu": 234.7, "t4": 5.86, "params": 25.9, "flops": 78.9},
            "l": {"size": 640, "map": 52.9, "cpu": 375.2, "t4": 9.06, "params": 43.7, "flops": 165.2},
            "x": {"size": 640, "map": 53.9, "cpu": 479.1, "t4": 14.37, "params": 68.2, "flops": 257.8},
        },
    },
    "YOLOv7": {
        "author": "Wang, Chien-Yao and Bochkovskiy, Alexey and Liao, Hong-Yuan Mark",
        "date": "2022-07-06",
        "arxiv": "https://arxiv.org/abs/2207.02696",
        "github": "https://github.com/WongKinYiu/yolov7",
        "docs": "https://docs.ultralytics.com/models/yolov7/",
        "performance": {
            "l": {"size": 640, "map": 51.4, "cpu": "", "t4": 6.84, "params": 36.9, "flops": 104.7},
            "x": {"size": 640, "map": 53.1, "cpu": "", "t4": 11.57, "params": 71.3, "flops": 189.9},
        },
    },
    "YOLOv6-3.0": {
        "author": "Chuyi Li and Lulu Li and Yifei Geng and Hongliang Jiang and Meng Cheng and Bo Zhang and Zaidan Ke and Xiaoming Xu and Xiangxiang Chu",
        "date": "2023-01-13",
        "arxiv": "https://arxiv.org/abs/2301.05586",
        "github": "https://github.com/meituan/YOLOv6",
        "docs": "https://docs.ultralytics.com/models/yolov6/",
        "performance": {
            "n": {"size": 640, "map": 37.5, "cpu": "", "t4": 1.17, "params": 4.7, "flops": 11.4},
            "s": {"size": 640, "map": 45.0, "cpu": "", "t4": 2.66, "params": 18.5, "flops": 45.3},
            "m": {"size": 640, "map": 50.0, "cpu": "", "t4": 5.28, "params": 34.9, "flops": 85.8},
            "l": {"size": 640, "map": 52.8, "cpu": "", "t4": 8.95, "params": 59.6, "flops": 150.7},
        },
    },
    "YOLOv5": {
        "author": "Glenn Jocher",
        "date": "2020-06-26",
        "arxiv": None,
        "github": "https://github.com/ultralytics/yolov5",
        "docs": "https://docs.ultralytics.com/models/yolov5/",
        "performance": {
            "n": {"size": 640, "map": 28.0, "cpu": 73.6, "t4": 1.12, "params": 2.6, "flops": 7.7},
            "s": {"size": 640, "map": 37.4, "cpu": 120.7, "t4": 1.92, "params": 9.1, "flops": 24.0},
            "m": {"size": 640, "map": 45.4, "cpu": 233.9, "t4": 4.03, "params": 25.1, "flops": 64.2},
            "l": {"size": 640, "map": 49.0, "cpu": 408.4, "t4": 6.61, "params": 53.2, "flops": 135.0},
            "x": {"size": 640, "map": 50.7, "cpu": 763.2, "t4": 11.89, "params": 97.2, "flops": 246.4},
        },
    },
    "PP-YOLOE+": {
        "author": "PaddlePaddle Authors",
        "date": "2022-04-02",
        "arxiv": "https://arxiv.org/abs/2203.16250",
        "github": "https://github.com/PaddlePaddle/PaddleDetection/",
        "docs": "https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.8.1/configs/ppyoloe/README.md",
        "performance": {
            "t": {"size": 640, "map": 39.9, "cpu": "", "t4": 2.84, "params": 4.85, "flops": 19.15},
            "s": {"size": 640, "map": 43.7, "cpu": "", "t4": 2.62, "params": 7.93, "flops": 17.36},
            "m": {"size": 640, "map": 49.8, "cpu": "", "t4": 5.56, "params": 23.43, "flops": 49.91},
            "l": {"size": 640, "map": 52.9, "cpu": "", "t4": 8.36, "params": 52.20, "flops": 110.07},
            "x": {"size": 640, "map": 54.7, "cpu": "", "t4": 14.3, "params": 98.42, "flops": 206.59},
        },
    },
    "DAMO-YOLO": {
        "author": "Xianzhe Xu and Yiqi Jiang and Weihua Chen and Yilun Huang and Yuan Zhang and Xiuyu Sun",
        "date": "2022-11-23",
        "arxiv": "https://arxiv.org/abs/2211.15444v2",
        "github": "https://github.com/tinyvision/DAMO-YOLO",
        "docs": "https://github.com/tinyvision/DAMO-YOLO/blob/master/README.md",
        "performance": {
            "t": {"size": 640, "map": 42.0, "cpu": "", "t4": 2.32, "params": 8.5, "flops": 18.1},
            "s": {"size": 640, "map": 46.0, "cpu": "", "t4": 3.45, "params": 16.3, "flops": 37.8},
            "m": {"size": 640, "map": 49.2, "cpu": "", "t4": 5.09, "params": 28.2, "flops": 61.8},
            "l": {"size": 640, "map": 50.8, "cpu": "", "t4": 7.18, "params": 42.1, "flops": 97.3},
        },
    },
    "YOLOX": {
        "author": "Zheng Ge and Songtao Liu and Feng Wang and Zeming Li and Jian Sun",
        "date": "2021-07-18",
        "arxiv": "https://arxiv.org/abs/2107.08430",
        "github": "https://github.com/Megvii-BaseDetection/YOLOX",
        "docs": "https://yolox.readthedocs.io/en/latest/",
        "performance": {
            "nano": {"size": 416, "map": 25.8, "cpu": "", "t4": "", "params": 0.91, "flops": 1.08},
            "tiny": {"size": 416, "map": 32.8, "cpu": "", "t4": "", "params": 5.06, "flops": 6.45},
            "s": {"size": 640, "map": 40.5, "cpu": "", "t4": 2.56, "params": 9.0, "flops": 26.8},
            "m": {"size": 640, "map": 46.9, "cpu": "", "t4": 5.43, "params": 25.3, "flops": 73.8},
            "l": {"size": 640, "map": 49.7, "cpu": "", "t4": 9.04, "params": 54.2, "flops": 155.6},
            "x": {"size": 640, "map": 51.1, "cpu": "", "t4": 16.1, "params": 99.1, "flops": 281.9},
        },
    },
    "RTDETRv2": {
        "author": "Wenyu Lv and Yian Zhao and Qinyao Chang and Kui Huang and Guanzhong Wang and Yi Liu",
        "date": "2023-04-17",
        "arxiv": "https://arxiv.org/abs/2304.08069",
        "github": "https://github.com/lyuwenyu/RT-DETR/tree/main/rtdetrv2_pytorch",
        "docs": "https://github.com/lyuwenyu/RT-DETR/tree/main/rtdetrv2_pytorch#readme",
        "performance": {
            "s": {"size": 640, "map": 48.1, "cpu": "", "t4": 5.03, "params": 20, "flops": 60},
            "m": {"size": 640, "map": 51.9, "cpu": "", "t4": 7.51, "params": 36, "flops": 100},
            "l": {"size": 640, "map": 53.4, "cpu": "", "t4": 9.76, "params": 42, "flops": 136},
            "x": {"size": 640, "map": 54.3, "cpu": "", "t4": 15.03, "params": 76, "flops": 259},
        },
    },
    "EfficientDet": {
        "author": "Mingxing Tan and Ruoming Pang and Quoc V. Le",
        "date": "2019-11-20",
        "arxiv": "https://arxiv.org/abs/1911.09070",
        "github": "https://github.com/google/automl/tree/master/efficientdet",
        "docs": "https://github.com/google/automl/tree/master/efficientdet#readme",
        "performance": {
            "d0": {"size": 640, "map": 34.6, "cpu": 10.2, "t4": 3.92, "params": 3.9, "flops": 2.54},
            "d1": {"size": 640, "map": 40.5, "cpu": 13.5, "t4": 7.31, "params": 6.6, "flops": 6.10},
            "d2": {"size": 640, "map": 43.0, "cpu": 17.7, "t4": 10.92, "params": 8.1, "flops": 11.0},
            "d3": {"size": 640, "map": 47.5, "cpu": 28.0, "t4": 19.59, "params": 12.0, "flops": 24.9},
            "d4": {"size": 640, "map": 49.7, "cpu": 42.8, "t4": 33.55, "params": 20.7, "flops": 55.2},
            "d5": {"size": 640, "map": 51.5, "cpu": 72.5, "t4": 67.86, "params": 33.7, "flops": 130.0},
            "d6": {"size": 640, "map": 52.6, "cpu": 92.8, "t4": 89.29, "params": 51.9, "flops": 226.0},
            "d7": {"size": 640, "map": 53.7, "cpu": 122.0, "t4": 128.07, "params": 51.9, "flops": 325.0},
        },
    },
    "Gold-YOLO": {
        "author": "Mingxing Tan and Ruoming Pang and Quoc V. Le",
        "date": "2023-09-20",
        "arxiv": "https://arxiv.org/abs/2309.11331",
        "github": "https://github.com/huawei-noah/Efficient-Computing/tree/master/Detection/Gold-YOLO",
        "docs": "https://github.com/huawei-noah/Efficient-Computing/blob/master/Detection/Gold-YOLO/README.md",
        "performance": {
            "n": {"size": 640, "map": 39.9, "cpu": "", "t4": 1.66, "params": 5.6, "flops": 12.1},
            "s": {"size": 640, "map": 46.4, "cpu": "", "t4": 3.43, "params": 21.5, "flops": 46.0},
            "m": {"size": 640, "map": 51.1, "cpu": "", "t4": 6.43, "params": 41.3, "flops": 87.5},
            "l": {"size": 640, "map": 53.3, "cpu": "", "t4": 10.64, "params": 75.1, "flops": 151.7},
        },
    },
    "D-FINE": {
        "author": "Yansong Peng and Hebei Li and Peixi Wu and Yueyi Zhang and Xiaoyan Sun and Feng Wu",
        "date": "2024-10-17",
        "arxiv": "https://arxiv.org/abs/2410.13842",
        "github": "https://github.com/Peterande/D-FINE",
        "docs": "https://github.com/Peterande/D-FINE/blob/master/README.md",
        "performance": {
            "n": {"size": 640, "map": 42.8, "cpu": "", "t4": 2.28, "params": 4, "flops": 7},
            "s": {"size": 640, "map": 48.5, "cpu": "", "t4": 4.19, "params": 10, "flops": 25},
            "m": {"size": 640, "map": 52.3, "cpu": "", "t4": 6.85, "params": 19, "flops": 57},
            "l": {"size": 640, "map": 54.0, "cpu": "", "t4": 9.50, "params": 31, "flops": 91},
            "x": {"size": 640, "map": 55.8, "cpu": "", "t4": 15.04, "params": 62, "flops": 202},
        },
    },
    "YOLO-World": {
        "author": "Cheng, Tianheng and Song, Lin and Ge, Yixiao and Liu, Wenyu and Wang, Xinggang and Shan, Ying",
        "date": "2024-01-30",
        "arxiv": "https://arxiv.org/abs/2401.17270",
        "github": "https://github.com/AILab-CVC/YOLO-World",
        "docs": "https://docs.ultralytics.com/models/yolo-world/",
        "performance": {
            "s": {"size": 640, "map": 46.1, "cpu": "", "t4": 3.46, "params": 12.7, "flops": 51.0},
            "m": {"size": 640, "map": 51.0, "cpu": "", "t4": 7.26, "params": 28.4, "flops": 110.5},
            "l": {"size": 640, "map": 53.9, "cpu": "", "t4": 11.00, "params": 46.8, "flops": 204.5},
            "x": {"size": 640, "map": 54.7, "cpu": "", "t4": 17.24, "params": 72.88, "flops": 309.3},
        },
    },
    "RTMDet": {
        "author": "Chengqi Lyu and Wenwei Zhang and Haian Huang and Yue Zhou and Yudong Wang and Yanyi Liu and Shilong Zhang and Kai Chen",
        "date": "2022-12-14",
        "arxiv": "https://arxiv.org/abs/2212.07784",
        "github": "https://github.com/open-mmlab/mmdetection/tree/3.x/configs/rtmdet",
        "docs": "https://github.com/open-mmlab/mmdetection/tree/3.x/configs/rtmdet#readme",
        "performance": {
            "t": {"size": 640, "map": 41.1, "cpu": "", "t4": 2.54, "params": 4.8, "flops": 8.1},
            "s": {"size": 640, "map": 44.6, "cpu": "", "t4": 3.18, "params": 8.89, "flops": 14.8},
            "m": {"size": 640, "map": 49.4, "cpu": "", "t4": 6.82, "params": 24.71, "flops": 39.27},
            "l": {"size": 640, "map": 51.5, "cpu": "", "t4": 11.06, "params": 52.3, "flops": 80.23},
            "x": {"size": 640, "map": 52.8, "cpu": "", "t4": 19.66, "params": 94.86, "flops": 141.67},
        },
    },
    "YOLO-NAS": {
        "author": "Aharon, Shay and {Louis-Dupont} and {Ofri Masad} and Yurkova, Kate and {Lotem Fridman} and {Lkdci} and Khvedchenya, Eugene and Rubin, Ran and Bagrov, Natan and Tymchenko, Borys and Keren, Tomer and Zhilko, Alexander and {Eran-Deci}",
        "date": "2023-05-03",
        "Arxiv": None,
        "github": "https://github.com/Deci-AI/super-gradients/blob/master/YOLONAS.md",
        "Documentation": "https://docs.ultralytics.com/models/yolo-nas/",
        "performance": {
            "s": {"size": 640, "map": 47.5, "cpu": "", "t4": 3.09, "params": 12.2, "flops": 32.8},
            "m": {"size": 640, "map": 51.6, "cpu": "", "t4": 6.07, "params": 31.9, "flops": 88.9},
            "l": {"size": 640, "map": 52.2, "cpu": "", "t4": 7.84, "params": 42.02, "flops": 121.09},
        },
    },
    "FCOS": {
        "author": "Tian, Zhi and Shen, Chunhua and Chen, Hao and He, Tong",
        "date": "2019-04-02",
        "Arxiv": "https://arxiv.org/abs/1904.01355",
        "github": "https://github.com/tianzhi0549/FCOS/",
        "Documentation": "https://github.com/tianzhi0549/FCOS/?tab=readme-ov-file#installation",
        "performance": {
            "R50": {"size": 800, "map": 36.6, "cpu": "", "t4": 15.18, "params": 32.3, "flops": 250.9},
            "R101": {"size": 800, "map": 39.1, "cpu": "", "t4": 18.91, "params": 51.28, "flops": 346.1},
        },
    },
    "SSD": {
        "author": "Liu, Wei and Anguelov, Dragomir and Erhan, Dumitru and Szegedy, Christian and Reed, Scott and Fu, Cheng-Yang and Berg, Alexander C.",
        "date": "2015-12-08",
        "Arxiv": "https://arxiv.org/abs/1512.02325",
        "github": "https://github.com/weiliu89/caffe/tree/ssd",
        "Documentation": "https://github.com/weiliu89/caffe/tree/ssd?tab=readme-ov-file#installation",
        "performance": {
            "300": {"size": 300, "map": 25.5, "cpu": "", "t4": 3.97, "params": 34.3, "flops": 68.7},
            "512": {"size": 512, "map": 29.5, "cpu": "", "t4": 8.96, "params": 36.0, "flops": 197.3},
        },
    },
    "RTDETRv3": {
        "author": "Shuo Wang and Chunlong Xia and Feng Lv and Yifeng Shi",
        "date": "2024-09-13",
        "arxiv": "https://arxiv.org/abs/2409.08475",
        "github": "https://github.com/clxia12/RT-DETRv3",
        "docs": "https://github.com/clxia12/RT-DETRv3/blob/main/README.md",
        "performance": {
            "s": {"size": 640, "map": 48.1, "cpu": "", "t4": 5.03, "params": 20, "flops": 60},
            "m": {"size": 640, "map": 49.9, "cpu": "", "t4": 7.51, "params": 36, "flops": 100},
            "l": {"size": 640, "map": 53.4, "cpu": "", "t4": 9.76, "params": 42, "flops": 136},
            "x": {"size": 640, "map": 54.6, "cpu": "", "t4": 15.03, "params": 76, "flops": 259},
        },
    },
    "LWDETR": {
        "author": "Qiang Chen and Xiangbo Su and Xinyu Zhang and Jian Wang and Jiahui Chen and Yunpeng Shen and Chuchu Han and Ziliang Chen and Weixiang Xu and Fanrong Li and Shan Zhang and Kun Yao and Errui Ding and Gang Zhang and Jingdong Wang",
        "date": "2024-06-05",
        "arxiv": "https://arxiv.org/abs/2406.03459",
        "github": "https://github.com/Atten4Vis/LW-DETR",
        "docs": "https://github.com/Atten4Vis/LW-DETR/blob/main/README.md",
        "performance": {
            "t": {"size": 640, "map": 42.6, "cpu": "", "t4": 2.56, "params": 12.1, "flops": 11.2},
            "s": {"size": 640, "map": 48.0, "cpu": "", "t4": 3.72, "params": 14.6, "flops": 16.6},
            "m": {"size": 640, "map": 52.5, "cpu": "", "t4": 6.59, "params": 28.2, "flops": 42.8},
            "l": {"size": 640, "map": 56.1, "cpu": "", "t4": 10.57, "params": 46.8, "flops": 71.6},
            "x": {"size": 640, "map": 58.3, "cpu": "", "t4": 22.29, "params": 118.0, "flops": 174.1},
        },
    },
}

if __name__ == "__main__":
    import json

    with open("model_data.json", "w") as f:
        json.dump(data, f)