Skip to content

Update train_CNN_olivettifaces.py #48

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Update train_CNN_olivettifaces.py #48

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 12 additions & 17 deletions DeepLearning Tutorials/FaceRecognition_CNN(olivettifaces)/train_CNN_olivettifaces.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
import numpy
from PIL import Image


import theano
import theano.tensor as T
from theano.tensor.signal import downsample
Expand Down Expand Up @@ -57,23 +58,17 @@ def load_data(dataset_path):
test_label[i]=label[i*10+9]

#将数据集定义成shared类型，才能将数据复制进GPU，利用GPU加速程序。
def shared_dataset(data_x, data_y, borrow=True):
shared_x = theano.shared(numpy.asarray(data_x,
dtype=theano.config.floatX),
borrow=borrow)
shared_y = theano.shared(numpy.asarray(data_y,
dtype=theano.config.floatX),
borrow=borrow)
return shared_x, T.cast(shared_y, 'int32')



train_set_x, train_set_y = shared_dataset(train_data,train_label)
valid_set_x, valid_set_y = shared_dataset(valid_data,valid_label)
test_set_x, test_set_y = shared_dataset(test_data,test_label)
rval = [(train_set_x, train_set_y), (valid_set_x, valid_set_y),
(test_set_x, test_set_y)]
return rval
def shared_dataset(data_x, data_y, borrow=True):
shared_x = theano.shared(np.asarray(data_x, dtype=np.float32), borrow=borrow)
shared_y = theano.shared(np.asarray(data_y, dtype=np.int32), borrow=borrow)
return shared_x, shared_y

train_set_x, train_set_y = shared_dataset(train_data, train_label)
valid_set_x, valid_set_y = shared_dataset(valid_data, valid_label)
test_set_x, test_set_y = shared_dataset(test_data, test_label)

rval = [(train_set_x, train_set_y), (valid_set_x, valid_set_y), (test_set_x, test_set_y)]
return rval



Expand Down