diff --git a/training_data/to_npy.py b/training_data/to_npy.py index b848b77..2400f90 100644 --- a/training_data/to_npy.py +++ b/training_data/to_npy.py @@ -18,7 +18,9 @@ for path in paths: temp = Image.open(path) #remove alpha channel if temp.mode=='RGBA': - temp = im.convert('RGB') + temp = temp.convert('RGB') + print ("no alpha") + print temp.mode keep = temp.copy() keep = np.array(keep) x.append(keep)