bcmi / Object-Shadow-Generation-Dataset-DESOBAv2

[CVPR 2024] The dataset, code, and model for our paper "Shadow Generation for Composite Image Using Diffusion Model", CVPR, 2024.
Apache License 2.0
99 stars 11 forks source link

Questions about model parameter initialization #7

Closed dorianzhang7 closed 5 months ago

dorianzhang7 commented 5 months ago

Hello. This work is based on the ControlNet method. The structure diagram of the paper shows that the parameters of UNet in the model are frozen. I would like to know how the parameters of UNet are initialized. Can you send me parameters or links if it is a pre-trained model? Thanks for your reply!

Narumi-Maria commented 5 months ago

We freeze parameter settings consistent with Controlnet; you can refer to https://github.com/lllyasviel/ControlNet.