YanzuoLu / CFLD

[CVPR 2024 Highlight] Coarse-to-Fine Latent Diffusion for Pose-Guided Person Image Synthesis
MIT License
165 stars 11 forks source link

about pose_encoder #10

Closed ButoneDream closed 6 months ago

ButoneDream commented 6 months ago

I want to use DWPOSE to replace OPENPOSE key points. Can I delete the weight of pose_encoder and retrain?

    del state_dict['pose_encoder.conv_in.weight']
    del state_dict['pose_encoder.conv_in.bias']
YanzuoLu commented 6 months ago

Of course that you can have a try.