search

zhengchen1999 / CAT

PyTorch code for our NeurIPS 2022 paper "Cross Aggregation Transformer for Image Restoration"
Apache License 2.0
116 stars 8 forks source link

question about the model #8

Open xinli2008 opened 8 months ago

xinli2008 commented 8 months ago

Given an original image and a mask image for the model, can the model generate a repaired image to complete the mask region? Best

zhengchen1999 commented 8 months ago

I think the pre-trained models we provided cannot do this. But, you can train a new model according to the training setting of the inpainting task, and it may be possible.

xinli2008 commented 8 months ago

thank you for your reply and i will try that!