diff --git a/README.md b/README.md index 37d39e3..2114257 100644 --- a/README.md +++ b/README.md @@ -45,21 +45,26 @@ Required resource structure: ``` resource/ ├── NeuralHaircut/ -│ ├── diffusion_prior/ -│ │ └── model.pt # Diffusion prior model +│ ├── pretrained_models/ +│ │ ├── diffusion_prior/ +│ │ │ └── dif_ckpt.pt # Diffusion prior model +│ │ └── strand_prior/ +│ │ └── strand_ckpt.pt # Strand prior model │ └── PIXIE/ -│ └── pixie_data/ # PIXIE model data +│ └── pixie_data/ +│ └── pixie_data/ # PIXIE model data ├── Matte-Anything/ -│ ├── sam_vit_h_4b8939.pth # SAM model -│ ├── groundingdino_swint_ogc.pth # GroundingDINO model -│ └── model.pth # Matte-Anything model +│ ├── model.pth # Matte-Anything model +│ ├── sam_vit_h_4b8939.pth # SAM model +│ └── groundingdino_swint_ogc.pth # GroundingDINO model ├── openpose/ │ └── models/ -│ ├── pose/ # OpenPose pose models -│ └── face/ # OpenPose face models +│ └── pose/ +│ └── coco/ +│ └── pose_iter_584000.caffemodel # OpenPose pose model └── hyperIQA/ └── pretrained/ - └── hyperIQA.pth # Image quality assessment model + └── hyperIQA.pth # Image quality assessment model ``` ## Getting Started diff --git a/README_CN.md b/README_CN.md index 0a47ae5..792d214 100644 --- a/README_CN.md +++ b/README_CN.md @@ -45,21 +45,26 @@ ``` resource/ ├── NeuralHaircut/ -│ ├── diffusion_prior/ -│ │ └── model.pt # 扩散先验模型 +│ ├── pretrained_models/ +│ │ ├── diffusion_prior/ +│ │ │ └── dif_ckpt.pt ## 扩散先验模型 +│ │ └── strand_prior/ +│ │ └── strand_ckpt.pt ## 股线先验模型 │ └── PIXIE/ -│ └── pixie_data/ # PIXIE模型数据 +│ └── pixie_data/ +│ └── pixie_data/ ## PIXIE模型数据 ├── Matte-Anything/ -│ ├── sam_vit_h_4b8939.pth # SAM模型 -│ ├── groundingdino_swint_ogc.pth # GroundingDINO模型 -│ └── model.pth # Matte-Anything模型 +│ ├── model.pth ## Matte-Anything模型 +│ ├── sam_vit_h_4b8939.pth ## SAM模型 +│ └── groundingdino_swint_ogc.pth ## GroundingDINO模型 ├── openpose/ │ └── models/ -│ ├── pose/ # OpenPose姿态模型 -│ └── face/ # OpenPose人脸模型 +│ └── pose/ +│ └── coco/ +│ └── pose_iter_584000.caffemodel ## OpenPose姿态模型 └── hyperIQA/ └── pretrained/ - └── hyperIQA.pth # 图像质量评估模型 + └── hyperIQA.pth ## 图像质量评估模型 ``` ## 环境配置