← all repositories

knazeri/edge-connect

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

2.6k stars Python Image · Video · Audio
edge-connect
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.