上传者: 42120997
|
上传时间: 2021-11-10 15:52:33
|
文件大小: 33KB
|
文件类型: -
Fusion_concat
该存储库包含级联融合和求和融合的代码。 融合是在两种不同类型的数据(RGB和IR热数据)之间完成的。
准备
首先,应同时下载SYSU-MM01数据集和RegDB数据集。 然后,应使用以下命令来准备两个数据集:
python pre_process_sysu_clean.py
python pre_process_regdb_clean.py
数据集必须位于此存储库的此位置:\
../Datasets/SYSU
../Datasets/RegDB
使用
发起训练的命令:
python train.py
参数:
--fusion:要融合的层(早期,layer1,layer2 ..,layer5,单峰) --fuse:融合类型(cat / sum) --fold:折数(0到4) --dataset:数据集名称(regdb / sysu) --reid:ReI