Fix imageReader, seems working with Validation dataset