remove alpha channel

2024-11-28 20:09:58 +00:00 · 2018-02-26 21:07:01 -05:00 · 2018-02-26 21:07:01 -05:00 · 6477410f06
commit 6477410f06
parent 7e8c8c2967
1 changed files with 3 additions and 0 deletions
--- a/training_data/to_npy.py
+++ b/training_data/to_npy.py
@ -16,6 +16,9 @@ for path in paths:
    #img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
    #x.append(img)
    temp = Image.open(path)
+    #remove alpha channel
+    if temp.mode=='RGBA':
+        temp = im.convert('RGB')
    keep = temp.copy()
    keep = np.array(keep)
    x.append(keep)