|
thon |
|
import torch |
|
import matplotlib.pyplot as plt |
|
import numpy as np |
|
fig, axes = plt.subplots(1, 4, figsize=(15, 5)) |
|
axes[0].imshow(image) |
|
axes[0].set_title('Original Image') |
|
mask_list = [masks[0][0][0].numpy(), masks[0][0][1].numpy(), masks[0][0][2].numpy()] |
|
for i, mask in enumerate(mask_list, start=1): |
|
overlayed_image = np.array(image).copy() |
|
overlayed_image[:,:,0] = np.where(mask == 1, 255, overlayed_image[:,:,0]) |
|
overlayed_image[:,:,1] = np.where(mask == 1, 0, overlayed_image[:,:,1]) |
|
overlayed_image[:,:,2] = np.where(mask == 1, 0, overlayed_image[:,:,2]) |
|
|
|
axes[i].imshow(overlayed_image) |
|
axes[i].set_title(f'Mask {i}') |
|
|
|
for ax in axes: |
|
ax.axis('off') |
|
plt.show() |
|
|
|
Box Prompting |
|
You can also do box prompting in a similar fashion to point prompting. |