knazeri/edge-connect
A two-stage adversarial model that generates missing image regions by first hallucinating edges and then completing the image.

Velocity · 7d
+1.0
★ / day
Trend
→steady
star history
EdgeConnect is an image inpainting method published at ICCV 2019 that follows a lines-first-then-color approach. The model uses a two-stage architecture: an edge generator hallucinates missing edges for damaged regions, and a subsequent image completion network fills in the missing pixels using these hallucinated edges as guidance. Both stages use adversarial training with GANs and are implemented in PyTorch.