python 3 version?

Hello, Been running this lovely code in google colab. However, python 2 is going away this month, and there are various dependencies (around scipy.misc.imread, for example) that cause issues in python 3. 

Does anyone already have a version of this code working under python 3?