FlashReFocus: Interactive Image Refocusing in Seconds

Project page · Paper · Code · Video

How to use. (1) Upload a photograph, or pick an example, and press Restore: one-step diffusion restores the all-in-focus image once. (2) Click the restored image or the depth map to focus; each click runs only the 1.3M-parameter renderer. (3) Turn the aperture, reshape the depth curve, or upload your own depth map; switch to Two focal planes (click two points) or Tilt-shift (click the row to keep sharp).

1. Input photograph

Long side (pixels)

512 1536

Detail weight ω

0: pixel LoRA only (most faithful); higher restores more detail.

0 1.5

Examples (Unsplash portraits and our own photographs)

Depth used for rendering (near is bright): click to focus

Depth curve γ (D → D^γ)

< 1 spreads out the near range, > 1 the far range; blur follows the new spacing.

0.4 2.5

Tilt-shift: half-height of the sharp band

0.02 0.3

Your own depth map (optional; near is bright, any size)

2. Restored all-in-focus image: click to focus

Focus mode

Two planes: the first click sets A, the second B. Tilt-shift: click a row.

One focal plane Two focal planes Tilt-shift

Aperture (blur strength K)

0: everything sharp; 1: the widest aperture seen in training; above 1 extrapolates.

0 1.5

Show focal markers on the result

3. Refocused