0

I want to train YOLOv5 on aws sagemaker also deploy the model on sagemaker itself,need to know about entrypoint python script as well. how can I build a pipeline for this?

Rohan Dhere
  • 13
  • 1
  • 5
  • Please edit the question to limit it to a specific problem with enough detail to identify an adequate answer. – Community Oct 06 '21 at 08:10

3 Answers3

1

You can find the YoloV5 on SageMaker and MLOps using Step Functions here.

pontiffs
  • 11
  • 2
0

This official AWS blog post has information on how to deploy YOLOv4. I wonder if you can use it as a guide and change the model to v5.

If not, there is a 3rd party implementation of YOLOv5 here.

Amir Imani
  • 3,118
  • 2
  • 22
  • 24
0

Please refer to link below. It has some example notebooks you can use to train the model on Sagemaker jupyter notebook.

https://github.com/aws-samples/amazon-sagemaker-train-and-deploy-yolov5

Cicilio
  • 413
  • 6
  • 12