# l代表長整型無符號變量
count = multiprocessing.Value('l', 0)
correct = multiprocessing.Value('l', 0)
lock = multiprocessing.Lock()

# 要運(yùn)行的函數(shù)為run()，lis為劃分給每個(gè)進(jìn)程的識別圖片列表
p = [multiprocessing.Process(target=run, args=(lock, i, count, correct)) for i in lis]

for i in p:
  i.start()
for i in p:
  i.join()

# 多個(gè)線程對共享變量進(jìn)行操作，加鎖
lock.acquire()
# 圖片已經(jīng)識別完畢，總數(shù)加一
count.value += 1
# 下面兩個(gè)if分別為識別正確的情況
if i[0] == 3 and dis >= threshold:
  correct.value += 1
  print("正確率：{0:.5f} 總數(shù)：{1} 正確數(shù)：{2} 錯(cuò)誤數(shù)：{3} 參數(shù)個(gè)數(shù)：{4} 向量夾角：{5:.5f} 圖片1：{6} 圖片2：{7}".format(correct.value/count.value, count.value, correct.value, count.value-correct.value, i[0], dis[0], i[1], i[2]))
  lock.release()
  continue
if i[0] == 4 and dis < threshold:
  correct.value += 1
  print("正確率：{0:.5f} 總數(shù)：{1} 正確數(shù)：{2} 錯(cuò)誤數(shù)：{3} 參數(shù)個(gè)數(shù)：{4} 向量夾角：{5:.5f} 圖片1：{6} 圖片2：{7}".format(correct.value/count.value, count.value, correct.value, count.value-correct.value, i[0], dis[0], i[1], i[2]))
  lock.release()
  continue
print("識別錯(cuò)誤：參數(shù)個(gè)數(shù)：{0} 向量夾角：{1} 圖片1：{2} 圖片2：{3}".format(i[0], dis[0], i[1], i[2]))
# 不要遺漏解鎖，否則進(jìn)程會死鎖
lock.release()

manager = multiprocessing.Manager()
# 這個(gè)是用來在多個(gè)進(jìn)程中間共享的字典
sync_dict = manager.dict()
# 這個(gè)是使用pickle序列化到文件中用的臨時(shí)字典變量
mem_dict = dict()

# 如果存在字典就載入，使用臨時(shí)字典中轉(zhuǎn)是因?yàn)閟ysnc_dict直接序列化會在下次加載時(shí)導(dǎo)致Manager在多個(gè)進(jìn)程中的連接出現(xiàn)問題
if os.path.exists("./muti_thread_mem.pkl"):
  with open("./muti_thread_mem.pkl", "rb") as f:
    mem_dict = pickle.load(f)
  for i in mem_dict:
    sync_dict[i] = mem_dict[i]

if i[1] in sync_dict:
 # 有記錄就直接讀取緩存
  encoding1 = sync_dict[i[1]]
else:
 # 沒記錄就計(jì)算出來再緩存一下
  face.file = Image.open(prefix+i[1])
  encoding1 = face.encodings
  sync_dict[i[1]] = encoding1
if i[2] in sync_dict:
  encoding2 = sync_dict[i[2]]
else:
  face.file = Image.open(prefix+i[2])
  encoding2 = face.encodings
  sync_dict[i[2]] = encoding2

with open("./muti_thread_mem.pkl", 'wb') as f:
  for k, v in enumerate(sync_dict):
    mem_dict[v] = sync_dict[v]
  if len(mem_dict):
   # 傳入的4就是想試一下Python新版本的特性而已，沒啥其他意思
    pickle.dump(mem_dict, f, 4)