def test_inference_other(self):
with self.test_session() as sess:
def d_cherry():
return w.dataset_paths([
'assets/cat.png',
])
def dm():
return w.dataset_paths([
'assets/empty.png',
])
def ds():
return w.dataset_paths([
'assets/cat-selection.png',
])
results = w.inference(sess, d_cherry, 1, dm, ds)
self.assertTupleEqual(results[0].shape, (1, 300, 300, 3))
def test_inference_voc(self):
with self.test_session() as sess:
dv = w.dataset_voc2012
results = w.inference(sess, dv, 1)
self.assertTupleEqual(results[0].shape, (4, 120, 120, 3))