Skip to content
Snippets Groups Projects
Commit 61535248 authored by Tamino Huxohl's avatar Tamino Huxohl
Browse files

add function to remove bed

parent ea5324c4
No related branches found
No related tags found
No related merge requests found
......@@ -34,6 +34,12 @@ def scale_points(points: List[List[int]], scale: float):
for j in range(len(points[i])):
points[i][j] = round(points[i][j] * scale)
def remove_bed(mu_map: np.ndarray, contour: np.ndarray):
_mu_map = mu_map.copy()
for i in range(_mu_map.shape[0]):
mu_map[i] = cv.drawContours(_mu_map[i], [bed_contour], -1, 0.0, -1)
return _mu_map
if __name__ == "__main__":
import argparse
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment