How to randomly change some pixels to black in any image within Pytorch?

1

How to randomly change some pixels to black in any image within Pytorch? Also, how to create a U-Net for inpainting?

2
  1. You could index the pixel location directly and set the desired pixel value (0 for a black pixel).
  2. GitHub seems to find some repositories for inpainting in PyTorch.
3

Any examples on how to do it on a regular JPG image?

4

Assuming your image is a numpy array (or you could transform it to one), this would work:

image_array = np.random.randint(0, 255, (3, 24, 24)).astype(np.uint8)
image_array[:, 17, 23] = 0
5

Can I use numpy within pytorch?

6

Yes, you can transform the tensor to a numpy array via tensor.numpy(), but would also be able to index the PyTorch tensor in the same way.

7

Thanks for the help! It worked.

How would I change a U-Net that segments to a U-Net that inpaints?

Best,

8

I don’t know the necessary changes and would take a look at some of the linked repositories.