Tensorflow example

[wjpearson]

Optical example for AA use in Tensorflow. Includes a script to load the GalaxyMNIST dataset as a tensorflow dataset.

[mb010]

I think if I merge this right now, then anyone who uses astroaugs will have to have both torch and tf installed. I will try to figure out a way to make that optional and then merge this?

My current plan is to remove any direct reference to data sets, and instead link them or provide them as auxiliary files which can be used, but wont be installed by pip (in the same way as the notebooks aren't installed).

I'm open to suggestions though.

Thanks for setting this up @wjpearson ! 🎉

[wjpearson]

I checked in a clean environment. I only installed: numpy, astropy, h5py, torch, torchvision (plus any dependencies they install to work). When importing AA it does not complain tensorflow is missing via the following:

>>> import sys
>>> sys.path.append('/home/pearsonw/dev/deep_learning/AstroAugmentations')
>>> import astroaugmentations as AA

So I don't think it requires tensorflow to use it.

But please double check.