Update modelling_magi.py

modelling_magi.py

@@ -67,7 +67,7 @@ class MagiModel(PreTrainedModel):
             character_detection_threshold=0.3,
             panel_detection_threshold=0.2,
             text_detection_threshold=0.25,
-            character_character_matching_threshold=0.7,
+            character_character_matching_threshold=0.65,
             text_character_matching_threshold=0.4,
         ):
         assert not self.config.disable_detections
@@ -454,6 +454,7 @@ class MagiModel(PreTrainedModel):
             char_ij_c2c = torch.cat([char_ij, c2c], dim=-1)
             character_character_affinities = self.character_character_matching_head(char_ij_c2c)
             character_character_affinities = rearrange(character_character_affinities, "(i j) 1 -> i j", i=char_i.shape[0])
+            character_character_affinities = (character_character_affinities + character_character_affinities.T) / 2
             if apply_sigmoid:
                 character_character_affinities = character_character_affinities.sigmoid()
             affinity_matrices.append(character_character_affinities)