GuHuangAI / DiffusionEdge

Code for AAAI 2024 paper: "DiffusionEdge: Diffusion Probabilistic Model for Crisp Edge Detection"
Apache License 2.0
236 stars 25 forks source link

the design of c_skip and c_out #12

Closed Maxwell-Zhao closed 8 months ago

Maxwell-Zhao commented 8 months ago

In the U-net, why you use this in the end? x1 = c_skip1 x_clone + c_out1 x1 x2 = c_skip2 x_clone + c_out2 x2

GuHuangAI commented 8 months ago

Please refer to EDM, and this design sightly affects the performance.