view scripts/imgbg_test.py @ 277:20ebc1f2a9fe

Use softmax for the output layer and rework the dset iterator stuff.
author Arnaud Bergeron <abergeron@gmail.com>
date Mon, 22 Mar 2010 16:37:34 -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