Mirrors/Track-Anything
1
0
Fork 0
mirror of https://github.com/gaomingqi/Track-Anything.git synced 2025-12-16 08:27:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix painter

Browse Source
This commit is contained in:
gaomingqi
2023-04-14 00:24:50 +08:00
parent cb491193d1
commit 5e67e9acff
2 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -4,3 +4,4 @@ docs/
*.pth *.pth
*.mp4 *.mp4
debug_images/ debug_images/
*.png

13
tools/painter.py
View File

@@ -150,9 +150,9 @@ def mask_painter(input_image, input_mask, mask_color=5, mask_alpha=0.7, contour_
contour_mask[contour_mask>0.5] = 1. contour_mask[contour_mask>0.5] = 1.
# paint mask # paint mask
painted_image = vis_add_mask(input_image, mask, mask_color, mask_alpha) painted_image = vis_add_mask(input_image.copy(), mask.copy(), mask_color, mask_alpha)
# paint contour # paint contour
painted_image = vis_add_mask(painted_image, 1-contour_mask, contour_color, 1) painted_image = vis_add_mask(painted_image.copy(), 1-contour_mask, contour_color, 1)
return painted_image return painted_image
@@ -166,10 +166,15 @@ if __name__ == '__main__':
mask_alpha = 0.7 mask_alpha = 0.7
contour_color = 1 contour_color = 1
contour_width = 5 contour_width = 5
# save
painted_image = Image.fromarray(input_image)
painted_image.save('images/original.png')
painted_image = mask_painter(input_image, input_mask, mask_color, mask_alpha, contour_color, contour_width) painted_image = mask_painter(input_image, input_mask, mask_color, mask_alpha, contour_color, contour_width)
# save # save
painted_image = Image.fromarray(painted_image) painted_image = Image.fromarray(input_image)
painted_image.save('images/mask_painter.png') painted_image.save('images/original1.png')
# example of point painter # example of point painter
input_image = np.array(Image.open('images/painter_input_image.jpg').convert('RGB')) input_image = np.array(Image.open('images/painter_input_image.jpg').convert('RGB'))