PYRA

Paper GitHub

Pyramid-focus-augmentation (PYRA): medical image segmentation with step-wise focus

We introduce a novel augmentation technique called pyramid-focus-augmentation which can be used to improve the performance of polyps segmentation task when we have a small dataset to train DL models. In addition to that, our method can focus doctors' attention to regions of polyps gradually. This technique can be applied to any segmentation task with any deep learning model using to do segmentation.

How to train your modle?

Sample output

Example input, input grids, converted groud truths, predicted mean and predicted standard deviation.

How to use PYRA Pytorch dataloader?

pip install pyra-pytorch
Please refer package documentation: Link