gistfile1.txt

from os import listdir
from pathlib import Path
from PIL import Image

root_dir = Path('images_xray', 'images')

covid_train_dir = Path(root_dir, 'train', 'covid')
noncovid_train_dir = Path(root_dir, 'train', 'noncovid')

covid_test_dir = Path(root_dir, 'test', 'covid')
noncovid_test_dir = Path(root_dir, 'test', 'noncovid')

dir_list = [covid_train_dir, noncovid_train_dir, covid_test_dir, noncovid_test_dir]

for workingdir in dir_list:
  for filename in workingdir:
    image = Image.open(filename)
    image.convert('RGB')
    image.save(Path(workingdir, filename))

clean image data covid classification