步态识别 GaitSet

介绍：

OpenGait：首个步态识别框架开源了！

开源地址：

https://github.com/ShiqiYu/OpenGait

复旦提出GaitSet算法，步态识别的重大突破！

时间0.28s

网络输入是分割好的二值图，黑色背景，白色人体。

应该要裁剪好的，网络输入是64*64的。本地跑起来了。

测试路径：

'dataset_path': "../test_part_data",

计算距离：

def cuda_dist(x, y):
    x = torch.from_numpy(x).cuda()
    y = torch.from_numpy(y).cuda()
    dist = torch.sum(x ** 2, 1).unsqueeze(1) + torch.sum(y ** 2, 1).unsqueeze(
        1).transpose(0, 1) - 2 * torch.matmul(x, y.transpose(0, 1))
    dis