mirror of
https://github.com/YaoFANGUK/video-subtitle-remover.git
synced 2026-02-27 14:14:44 +08:00
15 lines
466 B
Python
15 lines
466 B
Python
import matplotlib.pyplot as plt
|
|
from skimage import io
|
|
from skimage.transform import resize
|
|
|
|
from saicinpainting.evaluation.masks.mask import SegmentationMask
|
|
|
|
im = io.imread('imgs/ex4.jpg')
|
|
im = resize(im, (512, 1024), anti_aliasing=True)
|
|
mask_seg = SegmentationMask(num_variants_per_mask=10)
|
|
mask_examples = mask_seg.get_masks(im)
|
|
for i, example in enumerate(mask_examples):
|
|
plt.imshow(example)
|
|
plt.show()
|
|
plt.imsave(f'tmp/img_masks/{i}.png', example)
|