--- a +++ b/code/testDNN.lua @@ -0,0 +1,26 @@ +require 'torch'; +require 'nn'; + +cudaFlag = true + +if cudaFlag then + require 'cutorch'; + require 'cunn'; +end + +local folder = '/home/andrew/mitosis/data/mitosis-test/' +local batchSize = 200 + +dofile("data.lua") + +local classes, testClassList, imagePaths = getImagePaths(folder) + +--local net = torch.load('/home/andrew/mitosis/nets/dnn2_fullset_aug_30i_lr05_mini200_fcdropout.t7') +--local net = torch.load('/home/andrew/mitosis/nets/dnn2_fullset_aug_30i_lr05_mini200_weightdecay0001.t7') +local net = torch.load('/home/andrew/mitosis/data/nets/dnn2_fullset_aug_20i_lr05_lrd0005_m09_mini200_aeptgl.t7') +--local net = torch.load('/home/andrew/mitosis/data/nets/net.t7') +--local net = torch.load('/home/andrew/mitosis/data/nets/testNet.t7') + +-- test the network +dofile("test.lua") +test(net, classes, testClassList, imagePaths, batchSize)