训练GaitSet

训练GaitSet过程、问题及解决方法

1.数据集下载

下载CASIA-B数据集，解压。

2.数据集预处理

需要用pretreatment.py对数据集进行预处理，这个代码里有一处地方需要修改，在cut_pickle函数中保存图片时，他用了

scisc.imsave(save_path, img)

但是直接运行的话会出现保存不了的情况，并且最后还会报错，查找了scipy.misc.imsave和cv2.imwrite的区别后，发现前面读图片用了OpenCV，因此存图片也应该用OpenCV，因此修改成

cv2.imwrite(save_path, img)

3.训练

3.1 修改配置文件

首先要修改config.py中的一些参数，主要是dataset_path,CUDA_VISIBLE_DEVICES,这里的pid_num是指训练集中的人数，也要按照自己数据集中的情况修改。如下：

conf = {
    "WORK_PATH": "./work_test",
    "CUDA_VISIBLE_DEVICES": "0,1",
    "data": {
        'dataset_path': "/home1/datasets/CASIA-B/pretreatment/",
        'resolution': '64',
        'dataset': 'CASIA-B',
        # In CASIA-B, data of subject #5 is incomplete.
        # Thus, we ignore it in training.
        # For more detail, please refer to
        # function: utils.data_loader.load_data
        'pid_num': 124,
        'pid_shuffle': False,
    },
    "model": {
        'hidden_dim': 256,
        'lr': 1e-4,
        'hard_or_full_trip': 'full',
        'batch_size': (8, 16),
        'restore_iter': 0,
        'total_iter': 80000,
        'margin': 0.2,
        'num_workers': 3,
        'frame_num': 30,
        'model_name': 'GaitSet_Test',
    },
}

3.2 出现问题

1.第一个是numpy版本问题，运行train.py后报错如下：

ValueError: Object arrays cannot be loaded when allow_pickle=False

我第一次运行时numpy是1.17.0版本，然后回退到了1.16.2后，这个问题解决。
2.python版本问题
报错：

AttributeError: module 'random' has no attribute 'choices'

这个问题参考了GitHub上GaitSet中的issue15，换成python3.6

4.测试

运行test.py文件查看模型准确率，但是出现如下问题：

ValueError: need at least one array to concatenate