LeslieZhoa / tensorflow-MTCNN

人脸检测MTCNN算法,采用tensorflow框架编写,从理解到训练,中文注释完全,含测试和训练,支持摄像头
364 stars 141 forks source link

运行 gen_hard_example.py这个文件,跑了一晚还没才处理了800张照片?? #27

Closed xuecangqiuye closed 4 years ago

xuecangqiuye commented 4 years ago

你好,非常高兴能够看到这么通俗易懂的代码。 因为项目需要,我对代码进行了修改,删除了landmark识别,将网络的保存格式转为了pb模型,并用了tf.layers api进行了改写网络,加入了batch_nromal。

目前,程序运行至生成hard_example 12 阶段。 前面几张图片处理速度还比较快,大概3秒一张,但是处理到800多张的时候,却需要2分钟一张了,想问一下,有没有人遇到这个问题,可以一起讨论一下。

centos7 8 core 16 mem 1080ti

xuecangqiuye commented 4 years ago

1it [00:01, 1.60s/it](1432, 1024, 3) 2it [00:03, 1.66s/it](768, 1024, 3) 3it [00:05, 1.70s/it](682, 1024, 3) 4it [00:07, 1.74s/it](852, 1024, 3) 5it [00:09, 1.90s/it](931, 1024, 3) 6it [00:12, 2.15s/it](732, 1024, 3) 7it [00:14, 2.33s/it](683, 1024, 3) 8it [00:17, 2.46s/it](819, 1024, 3) 9it [00:20, 2.73s/it](768, 1024, 3)

速度会持续变慢,到800多张的时候,120s/it