Add ZITS inpainting model
royshil opened this issue · comments
Issue Type
Feature Request
OS
Windows
OS architecture
x86_64
Programming Language
C++
Framework
ONNX
Model name and Weights/Checkpoints URL
The ZITS is a continuation of MST (which already exists in this repo)
https://github.com/DQiaole/ZITS_inpainting
The weights (Places dataset) are downloadable here
https://drive.google.com/drive/folders/1Dg_6ZCAi0U3HzrYgXwr9nSaOLnPsf9n-?usp=sharing
There is also ZITS++ available https://github.com/ewrfcas/ZITS-PlusPlus
and the weights are available to download as well.
Description
ZITS (published CVPR 2022) is the successor of MST inpainting, and ZITS++ was published in TPAMI in May 2023.
Relevant Log Output
No response
URL or source code for simple inference testing code
No response