[teenbear]

Thanks for the write up! Just a note: at least in the pytorch collab there are missing includes (numpy and the imread function)

[eyvindn]

The second cell looks a section title ("Imports and Notebook Utilities"), but contains the definitions of these functions and the imports. Run this cell and I suspect things should work.