Upload folder using huggingface_hub

.gitignore

@@ -6,14 +6,24 @@ checkpoints
 
 
 
+#face parsing model (segmentation) Done
+Other_dependencies/face_parsing/79999_iter.pth
 
 
 
 
+# Expression model (For quantitative analysis only)  Not used
+Other_dependencies/face_recon/epoch_latest.pth
+eval_tool/Deep3DFaceRecon_pytorch_edit/BFM/*.mat
+
+# Arcface model 
+Other_dependencies/arcface/model_ir_se50.pth
 
 # pose model (For quantitative analysis only)
 Other_dependencies/Hopenet_pose/hopenet_robust_alpha1.pkl
 
+# Landmark detection model 
+Other_dependencies/DLIB_landmark_det/shape_predictor_68_face_landmarks.dat
 
 # Qantitative results
 Quantitative_Analysis/*
@@ -59,6 +69,7 @@ ttt*
 
 
 *.pt
+*.pth
 *.ckpt
 
 

download_checkpoints.py

@@ -21,5 +21,10 @@ def _download(repo_id, filename, local_path: Path) -> Path:
 
 
 _download("CompVis/stable-diffusion-v-1-4-original",SD14_filename, SD14_localpath)
+
 _download("scy639/UniBioTransfer",PRETRAIN_CKPT_PATH, ".")
 _download("scy639/UniBioTransfer",PRETRAIN_JSON_PATH, ".")
+
+_download("scy639/UniBioTransfer","Other_dependencies/arcface/model_ir_se50.pth", ".")
+_download("scy639/UniBioTransfer","Other_dependencies/DLIB_landmark_det/shape_predictor_68_face_landmarks.dat", ".")
+_download("scy639/UniBioTransfer","Other_dependencies/face_parsing/79999_iter.pth", ".")