prep_img_set.py

prep_img_set.py

import os
import Image
import glob
import numpy

def is_grey_scale(img_path="lena.jpg"):
    im = Image.open(img_path).convert('RGB')
    w,h = im.size
    for i in range(w):
        for j in range(h):
            r,g,b = im.getpixel((i,j))
            if r != g != b: return False
    return True


mainpath='../images/Expressionism'
symbpath='./Expressionism/images'
artists=os.listdir(mainpath)


# rename grayscale images
for artist in artists:
    images=glob.glob(mainpath + '/' + artist + '/*.jpg')
    for image in images:
        if is_grey_scale(image):
            os.rename(image, image + '.1')


# create symbolic links
counts = [len(glob.glob(mainpath + '/' + artist + '/*.jpg')) for artist in artists]
for idx,artist in enumerate(artists):
    if counts[idx] >= 30:
        os.symlink('../../' + mainpath + '/' + artist, symbpath + '/' + artist)