Changes for good coding style
@@ -72,11 +72,8 @@ class SelectiveSampler(Layer, InputBatchQueueRunner):
image_shapes = {
name: data[name].shape for name in self.window.names}
static_window_shapes = self.window.match_image_shapes(image_shapes)
- candidates = candidate_indices(
- static_window_shapes['label'],
-
- data['label'],
- self.constraint)
+ candidates = candidate_indices(static_window_shapes['label'],
+ data['label'], self.constraint)
if not np.sum(candidates) >= self.window.n_samples:
break
@@ -163,7 +160,8 @@ def candidate_indices(win_sizes, data, constraint):
else:
padding = padding + [[0, 0], ]
print(shape_ones, padding)
- final = np.pad(np.ones(shape_ones), np.asarray(padding, dtype=np.int32),
+ final = np.pad(np.ones(shape_ones), np.asarray(padding,
+ dtype=np.int32),
'constant')
new_win_size = np.copy(win_sizes)
# new_win_size[:N_SPATIAL] = win_sizes[0]/8
