model_list.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  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. 'CLIP_vit_base_patch16_224',
  13. 'CLIP_vit_large_patch14_224',
  14. 'ConvNeXt_tiny',
  15. 'MobileNetV2_x0_25',
  16. 'MobileNetV2_x0_5',
  17. 'MobileNetV2_x1_0',
  18. 'MobileNetV2_x1_5',
  19. 'MobileNetV2_x2_0',
  20. 'MobileNetV3_large_x0_35',
  21. 'MobileNetV3_large_x0_5',
  22. 'MobileNetV3_large_x0_75',
  23. 'MobileNetV3_large_x1_0',
  24. 'MobileNetV3_large_x1_25',
  25. 'MobileNetV3_small_x0_35',
  26. 'MobileNetV3_small_x0_5',
  27. 'MobileNetV3_small_x0_75',
  28. 'MobileNetV3_small_x1_0',
  29. 'MobileNetV3_small_x1_25',
  30. 'PP-HGNet_small',
  31. 'PP-HGNetV2-B0',
  32. 'PP-HGNetV2-B4',
  33. 'PP-HGNetV2-B6',
  34. 'PP-LCNet_x0_25',
  35. 'PP-LCNet_x0_35',
  36. 'PP-LCNet_x0_5',
  37. 'PP-LCNet_x0_75',
  38. 'PP-LCNet_x1_0',
  39. 'PP-LCNet_x1_5',
  40. 'PP-LCNet_x2_0',
  41. 'PP-LCNet_x2_5',
  42. 'ResNet101',
  43. 'ResNet152',
  44. 'ResNet18',
  45. 'ResNet34',
  46. 'ResNet50',
  47. 'SwinTransformer_base_patch4_window7_224',
  48. ]