convolutional_pose_machine.py 文件源码

def __call__(self, image, cmap, t):
        self.clear()
        h1 = self.stage1(image)
        h2 = self.branch(image)
        self.loss = F.mean_squared_error(h1, t)

        for name, _ in self.forward:
            f = getattr(self, name)
            h1 = f(h1, h2, cmap)
            self.loss += F.mean_squared_error(h1, t)

        return h1, self.loss