To expand on Piotr’s suggestion. The most common type of network used for this is U-Net, but you’ll probably have some difficulty to train that with the small size of your dataset unless you find some auxiliary data (need not necessarily be labeled, but from a similar domain).
Best regards
Thomas