Variationl auto-encoder on cell images