aws-samples / host-yolov8-on-sagemaker-endpoint

MIT No Attribution
35 stars 24 forks source link

Easy integration w AWS Lambda #22

Closed ArtemChemist closed 1 week ago

ArtemChemist commented 1 month ago
  1. Re-wrote endpoint invocation with boto3 library instead of sagemaker SDK. This way invoking YOLOv8 model from AWS lambda does not require installing an additional layer with sagemaker SDK and relies solely on pre-installed boto3. Using AWS lambda is a popular method for creating REST API for a model hosted on sagemaker, thus this update will be helpful for a wide audience. Given that interacting with YOLO often needs cv2 library, it is impossible to use sagmaker SDK in lambda, as total size of required libraries would exceed 250Mb limit.
  2. Updated tar structure to avoid invocation timeout issue mentioned in issue #21

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.