view scripts/imgbg_test.py @ 337:8d116d4a7593

Added convolutional RBM (ala Lee09) code, imported from my working dir elsewhere. Seems to work for one layer. No subsampling yet.
author fsavard
date Fri, 16 Apr 2010 16:05:55 -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