Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

a.py 938 B

You have to be logged in to leave a comment. Sign In
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
  1. from super_gradients.common.object_names import Models
  2. from super_gradients.training import models
  3. if __name__ == "__main__":
  4. model = models.get(Models.YOLO_NAS_POSE_S, num_classes=17)
  5. model.export(
  6. "2023_09_26_coco_yolo_nas_pose_s_1x640x640.onnx",
  7. input_image_shape=(640, 640),
  8. preprocessing=False,
  9. postprocessing=False,
  10. onnx_simplify=True,
  11. )
  12. model = models.get(Models.YOLO_NAS_POSE_M, num_classes=17)
  13. model.export(
  14. "2023_09_26_coco_yolo_nas_pose_m_1x640x640.onnx",
  15. input_image_shape=(640, 640),
  16. preprocessing=False,
  17. postprocessing=False,
  18. onnx_simplify=True,
  19. )
  20. model = models.get(Models.YOLO_NAS_POSE_L, num_classes=17)
  21. model.export(
  22. "2023_09_26_coco_yolo_nas_pose_l_1x640x640.onnx",
  23. input_image_shape=(640, 640),
  24. preprocessing=False,
  25. postprocessing=False,
  26. onnx_simplify=True,
  27. )
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...