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))