Requirements inference improvement

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

MIT No Attribution

35 stars 24 forks source link

Requirements inference improvement #18

Closed AnandSrikumar closed 3 months ago

AnandSrikumar commented 7 months ago

Issue #17

Description of changes: added nvgpu in requirements.txt, fixed output_fn of inference.py

In inference.py, we were trying to access result.keys, but result doesn't have an attribute keys, so I modified it. I accessed the boxes attribute and returned xyxy, xywh attributes of boxes.

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

shahromil16 commented 3 months ago

Ultralytics changes accessing keys to _keys. Latest PR merge uses this changed method.