view scripts/imgbg_test.py @ 272:f6d9b6b89c2a

ajouté : module de préparation de batches en fonction d'un ratio de classes
author Guillaume Sicard <guitch21@gmail.com>
date Mon, 22 Mar 2010 08:34:48 -0400
parents bb26c12bb9f6
children
line wrap: on
line source

#!/usr/bin/python

import Image, cPickle

f=open('/Tmp/image_net/filelist.pkl')
image_files = cPickle.load(f)
f.close()

for i in range(len(image_files)):
    filename = '/Tmp/image_net/' + image_files[i]
    try:
        image = Image.open(filename).convert('L')
    except:
        print filename