github上的facenet程式碼文件結構
阿新 • • 發佈:2018-11-27
文件結構
- 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中的繪製學習曲線工具