aws-samples / aws-panorama-samples

This repository has samples that demonstrate various aspects of AWS Panorama device and the Panorama SDK
https://docs.aws.amazon.com/panorama/
MIT No Attribution
82 stars 58 forks source link

Yolov5s fp16 #50

Open SuryaKari opened 2 years ago

SuryaKari commented 2 years ago

Issue #, if available:

Description of changes:

This Pull Request introduces an example of how to convert a YoloV5s model to half precision and use it on the Panorama device.

The change is inside the existing PT37_opengpu sample. This sample was only using an FP32 model but will now introduce the ability to convert this model to FP16 and then use it on the Panorama device. This model should take a dynamic input batch size.

We will be updating the inference latency metrics soon for this half precision model with various batch sizes

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