Add image dataset option to VisualInputModel
Allow choosing image dataset