torch-cryoeraser 0.1.0

Erase masked regions from cryo-EM images

torch-cryoeraser

torch-cryoeraser is a Python package for erasing local regions of cryo-EM images.

Image data in masked regions are replaced with noise matching local image statistics.

Installation

pip install torch-cryoeraser

Usage

import torch
import tifffile
from torch_cryoeraser import erase_region_2d

# load image and mask
image = tifffile.imread("image.tif")
mask = tifffile.imread("mask.tif")

# to torch tensor
image = torch.tensor(image)
mask = torch.tensor(mask)

# erase masked regions
erased_image = erase_region_2d(image=image, mask=mask)

for more info, please check the documentation.