Skip to content

Segmentor Docstrings

SegmentorPhotogrammetryCameraSet

Bases: PhotogrammetryCameraSet

Source code in geograypher/cameras/segmentor.py 
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
class SegmentorPhotogrammetryCameraSet(PhotogrammetryCameraSet):
    def __init__(
        self,
        base_camera_set: PhotogrammetryCameraSet,
        segmentor: Segmentor,
        dont_load_base_image: bool = True,
    ):
        """Wraps a camera set to provide segmented versions of the image

        Args:
            base_camera_set (PhotogrammetryCameraSet): The original camera set
            segmentor (Segmentor): A fully instantiated segmentor
        """
        self.base_camera_set = base_camera_set
        self.segmentor = segmentor
        self.dont_load_base_image = dont_load_base_image

        # This should allow all un-overridden methods to work as expected
        self.cameras = self.base_camera_set.cameras

    def get_image_by_index(self, index: int, image_scale: float = 1) -> np.ndarray:
        if self.dont_load_base_image:
            raw_image = None
        else:
            raw_image = self.base_camera_set.get_image_by_index(index, image_scale)
        image_filename = self.base_camera_set.get_image_filename(index, absolute=True)
        segmented_image = self.segmentor.segment_image(
            raw_image, filename=image_filename, image_scale=image_scale
        )
        return segmented_image

    def get_raw_image_by_index(self, index: int, image_scale: float = 1) -> np.ndarray:
        return self.base_camera_set.get_image_by_index(
            index=index, image_scale=image_scale
        )

    def get_subset_cameras(self, inds: typing.List[int]):
        subset_camera_set = deepcopy(self)
        subset_camera_set.cameras = [subset_camera_set.cameras[i] for i in inds]
        subset_camera_set.base_camera_set = (
            subset_camera_set.base_camera_set.get_subset_cameras(inds)
        )
        return subset_camera_set

    def n_image_channels(self) -> int:
        return self.segmentor.num_classes

Functions

__init__(base_camera_set, segmentor, dont_load_base_image=True)

Wraps a camera set to provide segmented versions of the image

Parameters:

Name Type Description Default
base_camera_set PhotogrammetryCameraSet

The original camera set

 required
segmentor Segmentor

A fully instantiated segmentor

 required
Source code in geograypher/cameras/segmentor.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
def __init__(
    self,
    base_camera_set: PhotogrammetryCameraSet,
    segmentor: Segmentor,
    dont_load_base_image: bool = True,
):
    """Wraps a camera set to provide segmented versions of the image

    Args:
        base_camera_set (PhotogrammetryCameraSet): The original camera set
        segmentor (Segmentor): A fully instantiated segmentor
    """
    self.base_camera_set = base_camera_set
    self.segmentor = segmentor
    self.dont_load_base_image = dont_load_base_image

    # This should allow all un-overridden methods to work as expected
    self.cameras = self.base_camera_set.cameras