layer {
name: "data"
type: "ImageData"
top: "data"
top: "label"
image_data_param {
source: "ImageList.txt"
batch_size: 8
new_height: 60
new_width: 60
}
include {
phase:
for x in range(0,224):
output = net.forward()
output_prob = output['prob'][0]
bOut.write(str(x) + "," + str(output_prob.argmax())+"\n")
print str(x) + ",", output_prob.argmax()
top_inds = output_prob.argsort()[::-1][:5]
print(zip(output_prob[top_inds]))