model_list.py 1.0 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. # copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License");
  4. # you may not use this file except in compliance with the License.
  5. # You may obtain a copy of the License at
  6. #
  7. # http://www.apache.org/licenses/LICENSE-2.0
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. MODELS = [
  15. 'PicoDet-L',
  16. 'PicoDet-S',
  17. 'PP-YOLOE_plus-L',
  18. 'PP-YOLOE_plus-M',
  19. 'PP-YOLOE_plus-S',
  20. 'PP-YOLOE_plus-X',
  21. 'RT-DETR-H',
  22. 'RT-DETR-L',
  23. 'RT-DETR-R18',
  24. 'RT-DETR-R50',
  25. 'RT-DETR-X',
  26. 'PicoDet_layout_1x',
  27. "YOLOv3-DarkNet53",
  28. "YOLOv3-MobileNetV3",
  29. "YOLOv3-ResNet50_vd_DCN",
  30. "YOLOX-L",
  31. "YOLOX-M",
  32. "YOLOX-N",
  33. "YOLOX-S",
  34. "YOLOX-T",
  35. "YOLOX-X",
  36. ]