1. 程式人生 > >github上的facenet程式碼文件結構

github上的facenet程式碼文件結構

文件結構

  • contributed/batch_represent.py 從圖片目錄中生成embeddings
  • contributed/cluster.py 人臉影象歸類
  • contributed/clustering.py 人臉匹配
  • contributed/export_embeddings.py 從圖片資料夾中匯出embeddings和標籤(numpy陣列格式)
  • contributed/face.py 人臉檢測和識別庫介面
  • contributed/predict.py
  • contributed/predict.py 實時人臉識別(獲取攝像頭圖片,並識別)
  • data/images 測試圖片資料夾
  • data/learning_rate_*.txt 遍歷次數與學習率對照表
  • data/pairs.txt
  • src/align/align_dataset_mtcnn.py 執行面對齊並將面部縮圖儲存在輸出目錄中
  • src/align/detect_face.py 基於多工級聯卷積神經網路的人臉檢測與對準
  • src/align/det*.npy detect_face.py 用到的資料
  • src/generative/models/dfc_vae.py 基於“深度特徵一致變分自動編碼器”的變分自動編碼器
  • src/generative/models/dfc_vae_large.py 基於“深度特徵一致變分自動編碼器”的變分自動編碼器(大圖片 128*128)
  • src/generative/models/dfc_vae_resnet.py 基於“深度特徵一致變分自動編碼器”的變分自動編碼器
  • src/generative/models/dfc_base.py 變分自動編碼器基類,包含編碼器和解碼器
  • src/generative/calculate_attribute_vectors.py 計算屬性向量
  • src/generative/modify_attribute.py 修改屬性向量
  • src/generative/train_vae.py 訓練變分自動編碼器
  • src/models/dummy.py 虛擬模型,用於測試
  • src/models/inception_resnet_v1.py Inception Resnet V1 網路結構
  • src/models/inception_resnet_v2.py Inception Resnet V2 網路結構
  • src/models/squeezenet.py squeezenet 網路結構
  • src/calculate_filtering_metrics.py 計算資料集的過濾指標並存儲在.hdf檔案中
  • src/classifier.py 使用自定義資料集訓練分類器
  • src/compare.py 進行面部對齊並計算與嵌入圖片的歐氏距離
  • src/decode_msceleb_dataset.py 解析msceleb資料集
  • src/download_and_extract.py
  • src/facenet.py 構建人臉識別網路
  • src/freeze_graph.py 凍結網路,輸出模型檔案
  • src/lfw.py 評估lfw資料集
  • src/train_softmax.py 使用交叉熵損失函式訓練
  • src/train_tripletloss.py 使用三元損失函式訓練
  • src/validate_on_lfw.py 在lfw資料集上驗證
  • test/*_test.py 測試程式碼
  • tmp
  • util/plot_learning_curves.m matlab中的繪製學習曲線工具