make the dataset loading generic #2
Description
Make sure the FF module can work with any type of image dataset with minimal changes to the way data is served to the model. This will lead us to use the existing encoding technique of overlaying the classification bit on image, or suggesting some new methodology. Ideally, we can create a class which is implemented by other child classes. And when this class is used in the code, we just call it encoding, and we can rapidly swap different encoding techniques.