search

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

MIT No Attribution
35 stars 24 forks source link

fix: update model packaging and inference #15

Closed charyin closed 3 months ago

charyin commented 8 months ago

Fixes #10 & Fixes #13

Description of changes:

  1. Create file structure recommended in the blog post.
  2. Minor refactoring of notebook sequence

Fixes #12

Description of changes:

  1. Update output_fn inline with ultralytics update.
  2. Update input_fn to remove hard coded model_dir which is available from the function parameter.

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