Update load_data.py

GZWQ · Nov 14, 2018 · 0a0f69d · 0a0f69d
1 parent a8f41d6
commit 0a0f69d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/load_data.py b/load_data.py
@@ -5,6 +5,7 @@
 nb_classes = 2
 from PIL import Image
 import matplotlib.pyplot as plt
+import os
 
 def cifar():
     print("Loading cifar10 data ...")