fourier-convolutionspytorchfourierinpainting-methodsdeep-neural-networksinpainting-algorithmdeep-learninginpaintingcomputer-visionimage-inpaintingcolab-notebookhigh-resolutioncolabgenerative-adversarial-networkscnngenerative-adversarial-networkganfourier-transform
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
466 B
14 lines
466 B
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)
|
|
|