learningcode's Introduction

CMC_rank

在 CMC.py 中通过 from CMC_rank.CMC_data_file import CMC.py 进行内部的计算

每当要加入新的数据时。

初始化 test_cmcno = []
test_cmcno = CMC(label_root, list_root, class_num)
test_cmcno = np.array(test_cmcno)
...
plt.plot(...)

insightface_pytorch_datasets_make

Insightface_pytorch版本的自定义数据准备过程

制作 .lst : 注意路径问题

python make_lst.py --dataset-dir D:\PycharmProjects\LearningProject\insightface_pytorch_datasets_make\EAR_train --list-file D:\PycharmProjects\LearningProject\insightface_pytorch_datasets_make\lst\train.lst --img-ext '.bmp'

创建property文件，注意不要有后缀

在文件中输入：class_num,112,112

class_num 代表类别个数

112，112 代表图片尺寸

生成rec & idx 文件

运行face2rec.py (同时注意：需要 face_image.py 和 face_preprocess.py文件) 也可以通过：

python face2rec.py --prefix D:\PycharmProjects\LearningProject\insightface_pytorch_datasets_make\dataset_file\lst --root D:\PycharmProjects\LearningProject\insightface_pytorch_datasets_make\dataset_file\EAR_train

的方式进行。

生成 pairs.txt 文件

python generate_image_pairs.py --data-dir D:\PycharmProjects\LearningProject\
insightface_pytorch_datasets_make\dataset_file\EAR_train --outputtxt D:\PycharmProjects\LearningProject\insightface_pytorch_datasets_make\dataset_file\lst\pairs.txt

生成 .bin 文件

--data-dir: .lst 文件路径

python lfw2pack.py --data-dir D:\PycharmProjects\LearningProject\insightface_pytorch_datasets_make\dataset_file\lst --output D:\PycharmProjects\
LearningProject\insightface_pytorch_datasets_make\dataset_file\lst\EAR.bin --num-samepairs 11000

Recommend Projects