Huawei_DIGIX_ImgRetri_Top2

2020 DIGIX GLOBAL AI CHALLENGE - Digital Device Image Retrieval - WEARE队 亚军

0. 环境&依赖

环境：

• Ubuntu 16.04
• CUDA 10.1 CUDNN 7.6.4
• 1080Ti or V100
• 内存128G

本项目所有实验均为单卡运行，使用1080Ti时batchsize改为 8（类）x 4（张）。

依赖：

# 训练环境
conda env create -f env.yaml
conda activate pytorch

# 后处理
cd indexing/PyRetri-master
python setup.py install

1. 数据预处理

# 训练数据生成
root_dir="./data" # 数据集所在目录
save_dir="./data" 
patch=640

cd features/utils
python convert_jpg2npy.py -root_dir $root_dir -save_dir $save_dir -patch $patch

2. 文件目录

本项目部分代码参考开源仓库 reid-strong-baseline、fast-reid、PyRetri

.
├── prepare.sh (依赖包安装、数据预处理)
├── features
│   ├── checkpoints (保存训练过程权重)
│   ├── configs     (配置文件)
│   ├── data
│   ├── engine
│   ├── exp         (特征保存目录)
│   ├── log
│   ├── model
│   ├── README.md
│   ├── solver
│   ├── tools
│   └── utils
├── indexing
│   ├── features
│   ├── index_configs (特征检索配置文件)
│   ├── index_tools
│   ├── PyRetri-master
│   ├── README.md
│   ├── result_tmp    (检索结果保存目录)
│   └── trans_index_json.sh (特征检索启动文件)
└── env.yaml

3. 算法说明

详细方案请见 zhihu