cvep

Computer Vision Experiment Platform

系统安装

本项目以京东AI的fast-reid为基础进行开发，重点扩展在智慧交通领域的应用。

首先安装所需软件：

pip install yacs
pip install pyyaml
sudo apt-get cython
pip install termcolor
pip install tabulate
pip install tensorboard

下载内置缺省数据集：到百度网盘下面下载Market1501数据集的Market-1501-v15.09.15.zip文件，将其解压到./datasets/目录下，目录结构如下所示：

datasets/
    Market-1501-v15.09.15/
        bounding_box_test/
        bounding_box_train/

运行缺省训练程序：

python ./tools/train_net.py --config-file ./configs/Market1501/bagtricks_R50.yml MODEL.DEVICE "cuda:3"

这里指定使用第四块GPU进行训练。

在某个目录下会有很多文件，其中绝大部分均属于同一类，但是有个别文件不属于同一类别。程序的任务就是找出不属于同一类别的图片挑出来。

程序的入口点为./tools/train_net.py，由于采用了一个比较复杂的架构，具体训练相关代码在fastreid/engine/train_loop.py中的SimpleTrainer.run_step方法中。