PINTO_model_zoo icon indicating copy to clipboard operation
PINTO_model_zoo copied to clipboard

Add ZITS inpainting model

Open royshil opened this issue 2 years ago • 0 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

royshil avatar Sep 20 '23 21:09 royshil