diff --git a/src/TrafficLightDetector/traffic_light_images.py b/src/TrafficLightDetector/traffic_light_images.py index 852a27d..c4be521 100644 --- a/src/TrafficLightDetector/traffic_light_images.py +++ b/src/TrafficLightDetector/traffic_light_images.py @@ -13,7 +13,7 @@ class TrafficLightDetectorImages(TrafficLightDetector): self._set_image(image) def _save_image(self) -> None: - cv2.imwrite(str(IMAGES_OUT_PATH.joinpath(self.path.name)), self.image_copy) + cv2.imwrite(str(IMAGES_OUT_PATH.joinpath(self.path.name)), self.image) def draw(self) -> None: self._draw_circle()