rahix/openflexure-microscope-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add random noise to simulation image

Browse source
This commit is contained in:
Julian Stirling 2025-07-24 16:17:42 +01:00
parent 5701225158
commit 31e3d15890
1 changed files with 18 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

25
src/openflexure_microscope_server/things/camera/simulation.py
View file

@ -31,6 +31,9 @@ RATIO = 0.2
# Some colour variation, for bg detect. # Some colour variation, for bg detect.
BG_COLOR = [220, 215, 217] BG_COLOR = [220, 215, 217]
# Random Number Generator
RNG = np.random.default_rng()
class SimulatedCamera(BaseCamera): class SimulatedCamera(BaseCamera):
"""A Thing that simulates a camera for testing.""" """A Thing that simulates a camera for testing."""
@ -108,14 +111,18 @@ class SimulatedCamera(BaseCamera):
We also generate a KD tree to rapidly find blobs in an image We also generate a KD tree to rapidly find blobs in an image
""" """
self.blobs = np.zeros((n_blobs, 3)) self.blobs = np.zeros((n_blobs, 3))
rng = np.random.default_rng()
w = np.max(self.glyph_shape) w = np.max(self.glyph_shape)
self.blobs[:, 0] = rng.uniform(w / 2, self.canvas_shape[0] - w / 2, n_blobs) self.blobs[:, 0] = RNG.uniform(w / 2, self.canvas_shape[0] - w / 2, n_blobs)
self.blobs[:, 1] = rng.uniform(w / 2, self.canvas_shape[1] - w / 2, n_blobs) self.blobs[:, 1] = RNG.uniform(w / 2, self.canvas_shape[1] - w / 2, n_blobs)
self.blobs[:, 2] = rng.choice(len(self.sprites), n_blobs) self.blobs[:, 2] = RNG.choice(len(self.sprites), n_blobs)
def generate_canvas(self): def generate_canvas(self):
"""Generate a canvas.""" """Generate a canvas.
Canvas is int16 so that random noise can be added to simulation image before
changing to unit8 to stop wrapping.
"""
self.canvas = np.ones(self.canvas_shape, dtype=np.int16) self.canvas = np.ones(self.canvas_shape, dtype=np.int16)
self.canvas[:, :, 0] *= BG_COLOR[0] self.canvas[:, :, 0] *= BG_COLOR[0]
self.canvas[:, :, 1] *= BG_COLOR[1] self.canvas[:, :, 1] *= BG_COLOR[1]
@ -128,7 +135,6 @@ class SimulatedCamera(BaseCamera):
] -= self.sprites[int(sprite_size_index)] ] -= self.sprites[int(sprite_size_index)]
self.canvas[self.canvas < 0] = 0 self.canvas[self.canvas < 0] = 0
self.canvas[self.canvas > 255] = 255 self.canvas[self.canvas > 255] = 255
self.canvas = self.canvas.astype(np.uint8)
def generate_image(self, pos: tuple[int, int, int]): def generate_image(self, pos: tuple[int, int, int]):
"""Generate an image with blobs based on supplied coordinates.""" """Generate an image with blobs based on supplied coordinates."""
@ -152,7 +158,12 @@ class SimulatedCamera(BaseCamera):
raise ValueError( raise ValueError(
f"Image shape {image.shape} does not match intended shape {self.shape}" f"Image shape {image.shape} does not match intended shape {self.shape}"
) )
return image
# Add noise and convert to uint8
image += RNG.normal(scale=2, size=self.shape).astype("int16")
image[image < 0] = 0
image[image > 255] = 255
return image.astype("uint8")
def attach_to_server( def attach_to_server(
self, server: lt.ThingServer, path: str, setting_storage_path: str self, server: lt.ThingServer, path: str, setting_storage_path: str