/LearningCode

备份一些自己的代码,电脑坏两次导致心态崩了

Primary LanguagePython

CMC_rank

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

每当要加入新的数据时。

  1. 初始化 test_cmcno = []
  2. test_cmcno = CMC(label_root, list_root, class_num)
  3. test_cmcno = np.array(test_cmcno)
  4. ...
  5. 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