def test_get_windows_lots(self): t = time.time() image = Image(width=640, height=480, classes=[], name='test') window_params = WindowParams() window_params.min_width = 10 window_params.stride = 8 window_params.aspect_ratios = [0.5, 1, 1.5] window_params.scales = 1. / 2 ** np.array([0, 0.5, 1, 1.5, 2]) print(window_params) windows = image.get_windows(window_params) time_passed = time.time() - t print("Generating windows took %.3f seconds" % time_passed) print(np.shape(windows)) print(windows[:10, :]) rand_ind = np.random.permutation(np.shape(windows)[0])[:10] print(windows[rand_ind, :])
def test_get_windows(self): image = Image(3,2,['whatever'],'test') window_params = WindowParams( min_width=2,stride=1,scales=[1,0.5],aspect_ratios=[0.5]) windows = image.get_windows(window_params) correct = np.array( [[ 0., 0., 2., 2.], [ 0., 0., 3., 2.], [ 1., 0., 3., 2.], [ 2., 0., 2., 2.], [ 0., 1., 2., 2.], [ 0., 1., 3., 2.], [ 1., 1., 3., 2.], [ 2., 1., 2., 2.], [ 0., 0., 3., 3.], [ 0., 0., 4., 3.]]) assert(windows.shape[0] > 0) print(windows) assert(np.all(correct == windows))