def runTest():
trBatchSize = 24
imgtransResize = 256
imgtransCrop = 224
print('Testing the trained model')
pathModel = '../models/model-m-11072018-225455-DenseNet201_loss = 0.001958067467107384_acc = 0.994636148354_best_acc.pth.tar'
# micro 3
# pathcsvData = ['/media/bmi/Varghese1/bla/cataract-2018-test/micro_3/test11/',
# '/media/bmi/Varghese1/bla/cataract-2018-test/micro_3/test12/',
# '/media/bmi/Varghese1/bla/cataract-2018-test/micro_3/test13/',
# '/media/bmi/Varghese1/bla/cataract-2018-test/micro_3/test14/',
# '/media/bmi/Varghese1/bla/cataract-2018-test/micro_3/test15/']
pathcsvData = [ #'/media/bmi/Varghese1/bla/cataract-2018-test/micro_2/test08/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_2/test09/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_2/test10/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_4/test16/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_4/test17/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_4/test18/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_4/test19/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_4/test20/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_5/test21/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_5/test22/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_5/test23/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_5/test24/',
#'/media/bmi/Varghese1/bla/cataract-2018-test/micro_5/test25/']
'/media/bmi/Varghese1/bla/cataract-2018-test/micro_1/test01/',
'/media/bmi/Varghese1/bla/cataract-2018-test/micro_1/test02/',
'/media/bmi/Varghese1/bla/cataract-2018-test/micro_1/test03/',
'/media/bmi/Varghese1/bla/cataract-2018-test/micro_1/test04/',
'/media/bmi/Varghese1/bla/cataract-2018-test/micro_1/test05/'
]
for path in pathcsvData:
Trainer.infer(model=DenseNet201(classCount=nclasses, isTrained=True),
pathcsvData=path,
pathModel=pathModel,
trBatchSize=trBatchSize,
transResize=imgtransResize,
transCrop=imgtransCrop)
