malaria_cell_CNN.py

data=[]
labels=[]

Parasitized=os.listdir("~/Downloads/cell_images/Parasitized/")
for a in Parasitized:
    try:
        image=cv2.imread("~/Downloads/cell_images/Parasitized/"+a)
        image_from_array = Image.fromarray(image, 'RGB')
        size_image = image_from_array.resize((50, 50))
        data.append(np.array(size_image))
        labels.append(0)
    except AttributeError as e:
        print(e)

Uninfected=os.listdir("~/Downloads/cell_images/Uninfected/")
for b in Uninfected:
    try:
        image=cv2.imread("~/Downloads/cell_images/Uninfected/"+b)
        image_from_array = Image.fromarray(image, 'RGB')
        size_image = image_from_array.resize((50, 50))
        data.append(np.array(size_image))
        labels.append(1)
    except AttributeError as e:
        print(e)