from optparse import OptionParser
parser = OptionParser()
parser.add_option("--test-data-dir", type="string", dest="test_data_dir")
parser.add_option("--classifier-file", type="string", dest="classifier_file")
options, args = parser.parse_args()
if options.test_data_dir == None or options.classifier_file == None:
print 'Please specify test data directory and classifier file'
sys.exit(-1)
models = d.get_models_from_dir('models/')
models_sdf = { x : s.sdf.from_file('models_sdf/'+x+'.npz') for x in models.keys()}
classifier = fern.fern_classifier.from_file(options.classifier_file)
num_scenes = json.load(open(options.test_data_dir+'dataset.json'))['num_scenes']
detected_scene_list = []
correct = 0.0
correct_object = 0.0
num_classificatons = 0
num_object_classificatons = 0
for i in range(num_scenes):
print 'Processing image', i
def __init__(self, options): self.options = options self.models = d.get_models_from_dir(options.models_dir) self.rw = self.get_render_window()
