分成3個階段介紹Dataset, Model Function 和 Train and Evaluate
因為需要在訓練同時做Data augmentation, 因此不使用 TFRecords 的方式