Update gligen/ldm/models/diffusion/loss.py

gligen/ldm/models/diffusion/loss.py

@@ -242,7 +242,7 @@ def caculate_loss_att_fixed_cnt(attn_maps_mid, attn_maps_up, attn_maps_down, bbo
                 
     return total_loss/obj_number
 
-def caculate_loss_LoCo(attn_maps_mid, attn_maps_up, attn_maps_down, bboxes, object_positions, t, res=16, smooth_att = True,sigma=0.5,kernel_size=3 ):
+def caculate_loss_LoCo(attn_maps_mid, attn_maps_up, attn_maps_down, bboxes, object_positions, t, res=16, smooth_att = False,sigma=0.5,kernel_size=3 ):
     attn16 = get_all_attention(attn_maps_mid, attn_maps_up, attn_maps_down, res)
     # attn32 = get_all_attention(attn_maps_mid, attn_maps_up, attn_maps_down, 32)
     # attn64 = get_all_attention(attn_maps_mid, attn_maps_up, attn_maps_down, 64)
@@ -290,7 +290,7 @@ def caculate_loss_LoCo(attn_maps_mid, attn_maps_up, attn_maps_down, bboxes, obje
                 # 选中物体对应位置（例如[6]）的map，然后reshape到[4, 16, 16]
                 
                 # print(attn_map[:, :, obj_position].shape)
-                ca_map_obj = attn_map[:, :, obj_position].sum(-1)
+                ca_map_obj = attn_map[:, :, obj_position].mean(-1)
                 
                 # print(ca_map_obj.shape)
                 if smooth_att: