Как рассчитать сходство/расстояние в сиамской сети (pytorch) ⇐ Python

[Anonymous]

Как именно вы рассчитываете сходство/расстояние в сиамской сети и после этого классифицируете их?
Это моя текущая попытка

Код: Выделить всё

class SiameseNetwork(nn.Module):
def __init__(self) -> None:
super().__init__()
self.resnet = torchvision.models.resnet18(num_classes=5)

def forward_once(self, item):
output = self.resnet(item)
return output

def forward(self, anchor, positive, negative):
output1 = self.forward_once(anchor)
output2 = self.forward_once(positive)
output3 = self.forward_once(negative)

return output1, output2, output3

Я использую resnet и TripletMarginLoss в качестве функции потерь, но я не понимаю, как вычислить сходство и классифицировать выходные данные после
также, если вы найдете если что-то не так с кодом, пожалуйста, сообщите мне.

Подробнее здесь: https://stackoverflow.com/questions/792 ... rk-pytorch