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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
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

    def get_subset_with_valid_segmentation(self) -> "SegmentorPhotogrammetryCameraSet":
        """Get a new camera set consisting of all images that have a valid segmentation result

        Returns:
            SegmentorPhotogrammetryCameraSet: The subset of cameras with valid segmentation
        """
        valid_inds = []
        for i in range(len(self)):
            try:
                # Try to get the segmented result
                self.get_image_by_index(i)
                # If successful, append it to the list of valid IDs
                valid_inds.append(i)
            except:
                pass
        # Return the valid subset
        return self.get_subset_cameras(valid_inds)

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

get_subset_with_valid_segmentation()

Get a new camera set consisting of all images that have a valid segmentation result

Returns:

Name Type Description
SegmentorPhotogrammetryCameraSet SegmentorPhotogrammetryCameraSet

The subset of cameras with valid segmentation
Source code in geograypher/cameras/segmentor.py 
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
def get_subset_with_valid_segmentation(self) -> "SegmentorPhotogrammetryCameraSet":
    """Get a new camera set consisting of all images that have a valid segmentation result

    Returns:
        SegmentorPhotogrammetryCameraSet: The subset of cameras with valid segmentation
    """
    valid_inds = []
    for i in range(len(self)):
        try:
            # Try to get the segmented result
            self.get_image_by_index(i)
            # If successful, append it to the list of valid IDs
            valid_inds.append(i)
        except:
            pass
    # Return the valid subset
    return self.get_subset_cameras(valid_inds)