view scripts/imgbg_test.py @ 189:0d0677773533

Fix bug where there would be a bunch of 0-length batches at the end under certain circumstances.
author Arnaud Bergeron <abergeron@gmail.com>
date Mon, 01 Mar 2010 17:06:49 -0500
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