torchio.data.sampler.uniform

Complexity

Total Complexity

2

Size/Duplication

Total Lines	17
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	UniformSampler.__init__()	0	2	1
A	UniformSampler.get_probability_map()	0	2	1

from typing import Union, Sequence
import torch
from .weighted import WeightedSampler


class UniformSampler(WeightedSampler):
    """Randomly extract patches from a volume with uniform probability.

    Args:
        patch_size: See :py:class:`~torchio.data.PatchSampler`.
    """
    def __init__(self, patch_size: Union[int, Sequence[int]]):
        super().__init__(patch_size)

    def get_probability_map(self, sample):
        return torch.ones(sample.shape)