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)