Converting YOLOv5 PyTorch Model Weights to TensorFlow Lite Format

Sergio Virahonda

Rate me:

5.00/5 (3 votes)

29 Jan 2021CPOL4 min read

39.2K

In this article we test a face mask detector on a regular computer.

Here we make our model understandable to TensorFlow Lite, the lightweight version of TensorFlow specially developed to run on small devices.

Views

Daily Counts

This article is part of the series 'AI on the Edge: Face Mask Detection ◁ Prev View All Next ▷

License

This article, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)

Written By

Sergio Virahonda

United States

Sergio Virahonda grew up in Venezuela where obtained a bachelor's degree in Telecommunications Engineering. He moved abroad 4 years ago and since then has been focused on building meaningful data science career. He's currently living in Argentina writing code as a freelance developer.

Converting YOLOv5 PyTorch Model Weights to TensorFlow Lite Format

Views

License

Comments and Discussions