Currently, datasets can contain images of any size. On load the images should be modified to be the same size as to work with all models.