Imported from https://github.com/Saran-nns/Edge-computing-with-tf