dltk package¶
dltk.utils module¶
-
class
dltk.utils.
SlidingWindow
(img_shape, window_shape, has_batch_dim=True, striding=None)[source]¶ Bases:
object
Sliding window iterator which produces slice objects to slice in a sliding window. This is useful for inference.
Constructs a sliding window iterator
Parameters: - img_shape (array_like) – shape of the image to slide over
- window_shape (array_like) – shape of the window to extract
- has_batch_dim (bool, optional) – flag to indicate whether a batch dimension is present
- striding (array_like, optional) – amount to move the window between each position
-
dltk.utils.
sliding_window_segmentation_inference
(session, ops_list, sample_dict, batch_size=1, striding=None)[source]¶ Utility function to perform sliding window inference for segmentation
Parameters: - session (tf.Session) – TensorFlow session to run ops with
- ops_list (array_like) – Operators to fetch assemble with sliding window
- sample_dict (dict) – Dictionary with tf.Placeholder keys mapping the
- placeholders to their respective input
- batch_size (int, optional) – Number of sliding windows to batch for calculation
- striding (array_like) – Striding of the sliding window. Can be used to adjust overlap etc.
Returns: - List of np.arrays corresponding to the assembled outputs of
ops_list
Return type: list