SegmentationBLWX / sssegmentation

SSSegmentation: An Open Source Supervised Semantic Segmentation Toolbox Based on PyTorch.
https://sssegmentation.readthedocs.io/en/latest/
Apache License 2.0
795 stars 107 forks source link

isnet:imagelevel.py #15

Closed shujunyy123 closed 2 years ago

shujunyy123 commented 3 years ago

imagelevel.py : 47: feats_il = self.correlate_net(x, torch.cat([x_global, x], dim=1))

isanet.py: 47:context = super(SelfAttentionBlock, self).forward(x, x)

is there any problem? bug?

CharlesPikachu commented 3 years ago

Sry, I cannot get you point. Could you describe your problem clearer?

shujunyy123 commented 3 years ago

(x, torch.cat([x_global, x], dim=1)) != (x, x) In debug mode, there’s a error .

CharlesPikachu commented 3 years ago

./models/base/selfattention.py stands for the self-attention-like structure, not self-attention in a narrow sense.