Skip to content

Commit

Permalink
colour update
Browse files Browse the repository at this point in the history
Signed-off-by: Ken Haagh <[email protected]>
  • Loading branch information
KenH2 committed Dec 14, 2023
1 parent e379323 commit 48ffbf3
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions people_tracking/src/people_tracking/person_detection.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,6 +110,8 @@ def process_latest_image(self):
# Import RGB Image
cv_image = self.bridge.imgmsg_to_cv2(latest_image, desired_encoding='passthrough')
cv_image = cv2.GaussianBlur(cv_image, (5, 5), 0)
if not laptop:
cv_image = cv2.cvtColor(cv_image, cv2.COLOR_RGB2BGR)

# Save Image
if save_data:
Expand All @@ -127,6 +129,8 @@ def process_latest_image(self):
depth_image = self.request_depth_image(self.latest_image_time).image
cv_depth_image = self.bridge.imgmsg_to_cv2(depth_image, desired_encoding='passthrough')
cv_depth_image = cv2.GaussianBlur(cv_depth_image, (5, 5), 0)
if not laptop:
cv_depth_image = cv2.cvtColor(cv_depth_image, cv2.COLOR_RGB2BGR)

if save_data:
cv2.imwrite(f"{save_path}{batch_nr}_depth.jpg", cv_depth_image)
Expand Down

0 comments on commit 48ffbf3

Please sign in to comment.