-
Notifications
You must be signed in to change notification settings - Fork 1
/
test.py
67 lines (43 loc) · 1.71 KB
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
from utils.config import process_config_test
from models.resnetRGB_model import ResNetRGBModel
from trainer.resnetRGB_trainer import ResNetRGBTrainer
from models.Persistence_model import PersistenceModel
from trainer.Persistence_trainer import PersistenceTrainer
from models.resnetCombined_model import ResNetCombinedModel
from trainer.resnetCombined_trainer import ResNetCombinedTrainer, ResNetCombinedTest
from utils.evaluate import test_combined, test_rgb, test_persistence
import numpy as np
from sklearn import metrics
import sys, os
import argparse
from data_loader.combined_data_loader import CombinedTestData
from keras import Model
import cPickle as pickle
def main():
argparser = argparse.ArgumentParser(description=__doc__)
argparser.add_argument('--test_dir', default='', help='Path to Test Directory')
argparser.add_argument('--config_dir', default='', help='Path to Config Directory (Containing model checkpoints and config file)')
args = argparser.parse_args()
if args.__dict__['test_dir'] == '':
print '-'*90
print 'Provide path to test folder: --test_dir (Should have directories : malignant and benign)'
print '-'*90
sys.exit(0)
if args.__dict__['config_dir'] == '':
print '-'*90
print 'Provide path to config folder: --config_dir'
print '-'*90
sys.exit(0)
config = process_config_test(args)
print '\n', config, '\n'
if config.rgb:
print 'Testing RGB model'
test_rgb(config)
elif config.combined:
print 'Testing Combined model'
test_combined(config)
else:
print 'Testing Persistence model'
test_persistence(config)
if __name__ == "__main__":
main()