davidsandberg / facenet

Face recognition using Tensorflow
MIT License
13.61k stars 4.8k forks source link

Performance issues in the program #1208

Open DLPerf opened 2 years ago

DLPerf commented 2 years ago

Hello,I found a performance issue in the definition of load_and_align_data , davidsandberg/facenet/blob/master/contributed/export_embeddings.py, sess = tf.Session was repeatedly called and was not closed. I think it will increase the efficiency and avoid out of memory if you close this session after using it.

Here are two files to support this issue,support1 and support2

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.