model_list.py 561 B

123456789101112131415161718192021222324
  1. # !/usr/bin/env python3
  2. # -*- coding: UTF-8 -*-
  3. ################################################################################
  4. #
  5. # Copyright (c) 2024 Baidu.com, Inc. All Rights Reserved
  6. #
  7. ################################################################################
  8. """
  9. Author: PaddlePaddle Authors
  10. """
  11. MODELS = [
  12. 'PicoDet-L',
  13. 'PicoDet-S',
  14. 'PP-YOLOE_plus-L',
  15. 'PP-YOLOE_plus-M',
  16. 'PP-YOLOE_plus-S',
  17. 'PP-YOLOE_plus-X',
  18. 'RT-DETR-H',
  19. 'RT-DETR-L',
  20. 'RT-DETR-R18',
  21. 'RT-DETR-R50',
  22. 'RT-DETR-X',
  23. 'PicoDet_layout_1x',
  24. ]