(1)确保本地环境对GPU的支持,当数据集的制作需要tensorflow里,需要安装gpu版本的tensorflow

 测试脚本:https://github.com/dualven/insightface/blob/master/src/align/tensorflowTest.py

安装脚本: conda install tensorflow-gpu

(2) 修改align脚本align_gbcom.py

使用GPU制作数据集

放开gpu限制。

正常情况下,程序启动有如下 打印,与GPU相关:

使用GPU制作数据集 

使用GPU制作数据集 

(3)align命令

python align_gbcom.py --input-dir /data/gbcomFaceBase/images/ --output-dir ../../datasets/gbcom2020-2 --image-size 112,112

说明:

https://github.com/dualven/insightface/blob/master/src/common/face_image.py

https://github.com/dualven/insightface/blob/master/src/align/align_gbcom.py

align_gbcom 调用 face_image。

生成了lst 

针对我们目前的目录结构进行原始数据处理

(4) property

   3810,112,112;

copy lst to rain-gbcom2020/train.lst

(5) src/data/

python face2rec2.py  ../../datasets/train-gbcom2020/

生成train.rec ,index

(6) pairs.txt

python generate_image_pairs.py --data-dir ../../datasets/gbcom2020-2 --outputtxt ../../datasets/train-gbcom2020/pairs.txt  --num-samepairs 5000

5000

5000

 

(7) src/data

 python lfw2pack.py --data-dir ../../datasets/train-gbcom2020/ --output ../../datasets/train-gbcom2020/gbcom.bin --num_samepairs 5000

生成 :

使用GPU制作数据集

 

相关文章:

  • 2021-12-06
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-01-12
  • 2021-12-13
  • 2021-05-30
  • 2021-06-20
猜你喜欢
  • 2022-12-23
  • 2021-08-14
  • 2022-12-23
  • 2021-12-26
  • 2021-04-25
  • 2021-12-18
  • 2021-07-24
相关资源
相似解决方案