fake = "Spoof"
#%%
testSensor = sensor.replace("Train","Test")
testDB = os.path.join(options.testData, year, 'Testing', testSensor)
#%%
for tfold in tfolders:
if tfold is not 'Live':
testDBFolder = os.path.join(testDB, fake, tfold)
else:
testDBFolder = os.path.join(testDB, tfold)
fileList = ut.search(testDBFolder, ['.png', '.bmp'])
saveResultFolder = os.path.join(testFolder, tfold)
for imgName in fileList:
img = caffe.io.load_image(imgName)
patch, patchCoor = tu.ExtractPatch(img, patchSize)
patchNum = len(patch)
net.blobs['data'].reshape(patchNum,3, patchSize, patchSize)
for i, pat in enumerate(patch):
transformed_img = transformer.preprocess('data', pat)
net.blobs['data'].data[i,...] = transformed_img
out = net.forward()
fc12 = net.blobs['fc12'].data
refc12 = fc12.argmax(axis=1)
result = []
for i, ll in enumerate(refc12):
result.append((patchCoor[i], ll, fc12[i, ll]))
ListToPickle(result, saveResultFolder, imgName.split('/')[-1][:-4])
print(tfold+' is finished')
# ListToTxtSave(saveResultFolder, imgName.split('.')[0]+'.txt')
#
